题目:
根据 逆波兰表示法,求表达式的值。
有效的运算符包括 +, -, *, / 。每个运算对象可以是整数,也可以是另一个逆波兰表达式。
说明:
整数除法只保留整数部分。
给定逆波兰表达式总是有效的。换句话说,表达式总会得出有效数值且不存在除数为 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
思路:借助栈
建立numStack 数字栈
- 每次从tokens里拿出一个元素
- 如果是数字:压到 numStack 数字栈里
- 如果是符号:从numStack 数字栈弹出2个数,计算在这个符号下的值,压入numStack 数字栈
- 重复上述过程直到tokens为空,numStack 数字栈剩下的数字就是结果
代码:
class Solution:
def evalRPN(self, tokens: List[str]) -> int:
if not tokens:
return
numStack = []
while tokens:
mark = tokens.pop(0)
if mark in "+-*/":
num1 = numStack.pop()
num2 = numStack.pop()
if mark == "+":
numStack.append(int(num2) + int(num1))
elif mark == "-":
numStack.append(int(num2) - int(num1))
elif mark == "*":
numStack.append(int(num2) * int(num1))
else:
numStack.append(int(num2) / int(num1))
else:
numStack.append(mark)
return int(numStack[0])