参考《大话数据结构》一书4.9节,Python下基于栈和逆波兰算法实现四则运算。
- 栈的实现使用顺序结构,即基于列表实现,参考本人博客Python实现顺序结构栈Sequence Stack
- 中缀转换后缀采用逆波兰算法,参考本人博客Python利用栈将中缀表达式转换为后缀表达式(四则运算)。
计算代码如下所示(transform函数见上述 2.Python利用栈将中缀表达式转换为后缀表达式(四则运算)):
def arithmetic(expression):
signs = '+-*/'
symbol = SeqStack(size=20)
expression = expression.split(' ') # 将后缀表达式分隔为数字和符号
for ch in expression:
if ch not in signs:
symbol.push(ch)
else:
b = symbol.pop() # 首先弹出的是符号右侧的数
a = symbol.pop()
c = eval(a + ch + b) # eval用于计算字符串中的表达式
symbol.push(str(c))
return symbol.pop()
def main():
infix = '1+((2+3)*4)-5'
infix = '42+21*2-(5-3)*4/2'
suffix = transform(infix)
print(suffix)
print(arithmetic(suffix))
if __name__ == '__main__':
main()