题目描述
根据 逆波兰表示法,求表达式的值。
有效的算符包括 +、-、*、/ 。每个运算对象可以是整数,也可以是另一个逆波兰表达式。
注意 两个整数之间的除法只保留整数部分。
可以保证给定的逆波兰表达式总是有效的。换句话说,表达式总会得出有效数值且不存在除数为 0 的情况。
示例 1:
输入:tokens = [“2”,“1”,“+”,“3”,“*”]
输出:9
解释:该算式转化为常见的中缀算术表达式为:((2 + 1) * 3) = 9
示例 2:
输入:tokens = [“4”,“13”,“5”,“/”,“+”]
输出:6
解释:该算式转化为常见的中缀算术表达式为:(4 + (13 / 5)) = 6
示例 3:
输入:tokens = [“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,
tokens = [“2”,“1”,“+”,“3”,“*”]
建立栈stack = []
遇到数字就进栈, 遇到计算符就计算倒数第二个和倒数第一个之间的数值结果,然后将该结果进栈。
- 遇到2进栈,遇到1进栈。 此时栈里 stack = [2,1]
- 遇到 ‘+’ 号计算符 将前两个出栈 做加运算,2+1=3
- 将计算结果进栈 此时栈里 stack = [3]
- 在遇到3 进栈 此时栈里 stack = [3,3]
- 遇到 ‘*’ 取出栈里的 3 * 3 = 9 ,在进栈 stack = [9]
这就是整体过程。
只有一个要注意的点,就是 运算顺序是 倒数第二个和倒数第一个的运算,也就是 stack[-2] 与 stack[-1] 在减法的除法的时候尤其注意。
完整代码
class Solution:
def evalRPN(self, tokens: List[str]) -> int:
stack = []
for i in tokens:
if i == '+':
stack.append(int(stack.pop()) + int(stack.pop()))
elif i == '-':
temp = int(stack[-2]) - int(stack[-1])
stack.pop()
stack.pop()
stack.append(temp)
elif i == '*':
stack.append(int(stack.pop()) * int(stack.pop()))
elif i == '/':
stack.append(1/int(stack.pop()) * int(stack.pop()))
else:
stack.append(i)
return int(stack[0])