根据 逆波兰表示法,求表达式的值。
有效的运算符包括 + , - , * , / 。每个运算对象可以是整数,也可以是另一个逆波兰表达式。
说明:
整数除法只保留整数部分。 给定逆波兰表达式总是有效的。换句话说,表达式总会得出有效数值且不存在除数为 0 的情况。
示例 1:
- 输入: ["2", "1", "+", "3", " * "]
- 输出: 9
- 解释: 该算式转化为常见的中缀算术表达式为:((2 + 1) * 3) = 9
示例 2:
- 输入: ["4", "13", "5", "/", "+"]
- 输出: 6
- 解释: 该算式转化为常见的中缀算术表达式为:(4 + (13 / 5)) = 6
示例 3:
-
输入: ["10", "6", "9", "3", "+", "-11", " * ", "/", " * ", "17", "+", "5", "+"]
-
输出: 22
-
解释:该算式转化为常见的中缀算术表达式为:
((10 * (6 / ((9 + 3) * -11))) + 17) + 5 = ((10 * (6 / (12 * -11))) + 17) + 5 = ((10 * (6 / -132)) + 17) + 5 = ((10 * 0) + 17) + 5 = (0 + 17) + 5 = 17 + 5 = 22
逆波兰表达式:是一种后缀表达式,所谓后缀就是指运算符写在后面。
平常使用的算式则是一种中缀表达式,如 ( 1 + 2 ) * ( 3 + 4 ) 。
该算式的逆波兰表达式写法为 ( ( 1 2 + ) ( 3 4 + ) * ) 。
逆波兰表达式主要有以下两个优点:
-
去掉括号后表达式无歧义,上式即便写成 1 2 + 3 4 + * 也可以依据次序计算出正确结果。
-
适合用栈操作运算:遇到数字则入栈;遇到运算符则取出栈顶两个数字进行计算,并将结果压入栈中。
我的思路:用stack
看到数字,就加入stack
看到运算符,就取出stack尾部的2个元素,根据运算符来进行相对应的运算,然后把运算结果加入stack。
重复这个过程,直到stack为空,就返回最后一次运算的结果。
题外话:
我们平常使用的表达式大多是中缀表达式,因为它们符合我们的阅读习惯。然而,对计算机来说,中缀表达式并不方便处理。
例如:6 - 8 * 2
是一个中缀表达式。计算机从左到右扫描时,遇到 8
后需要判断接下来的运算符,并比较运算符的优先级。然后,它需要先计算 8 * 2
,再回头处理 6 -
。这个过程既复杂又低效。
如果将中缀表达式转换为后缀表达式,即 ["6", "8", "2", "*", "-"]
,情况会简单得多。计算机可以利用栈来顺序处理这些元素,无需考虑运算符优先级,也不必回退。这样一来,后缀表达式对计算机处理起来更高效。
这个例子不仅仅是一道算法题,还展示了计算机在处理表达式时的独特思维方式。
注意:stack是先弹出最末尾的元素。所以,num2=res.pop( ), num1=res.pop( ), 然后num1与num2进行运算。
代码:
我写的:
class Solution:
def evalRPN(self, tokens: List[str]) -> int:
res = []
for i in tokens:
if i.isdigit():
res.append(float(i))
elif i == "+":
num2 = res.pop()
num1 = res.pop()
res.append(num1 + num2)
elif i == "-":
num2 = res.pop()
num1 = res.pop()
res.append(num1 - num2)
elif i == "*":
num2 = res.pop()
num1 = res.pop()
res.append(num1 * num2)
else:
num2 = res.pop()
num1 = res.pop()
res.append(num1 // num2)
return res.pop()
报错了。没太看出来是哪里有错误。欢迎大家在评论区留言指正!
正确代码:
from operator import add, sub, mul
class Solution:
op_map = {'+':add, '-':sub, '*':mul, '/': lambda x,y: int(x/y)}
def evalRPN(self, tokens: List[str]) -> int:
res = []
for i in tokens:
if i not in {'+','-','*','/'}:
res.append(int(i))
else:
num2 = res.pop()
num1 = res.pop()
res.append(self.op_map[i](num1,num2))
return res.pop()
注意:lamda x,y: int(x/y) 这个写法
if i not in {'+','-','*','/'} 是 { } 而不是 [ ]
op_map是一个dictionary,可以通过key,去找到key对应的value。也就是,key对应四个运算符的写法,其对应的value是真正的运算符号。
self.op_map[token] 这样,就把真正的运算符号找到了。