Problem: 150. 逆波兰表达式求值
题目
输入一个根据逆波兰表达式表示的四则运算表达式,输出计算结果。注意除法是向零取整。
示例1:输入:tokens = [“2”,“1”,“+”,“3”,““]; 输出: 9
示例2:输入:tokens = [“10”,“6”,“9”,“3”,”+“,”-11","”,“/”,“*”,“17”,“+”,“5”,“+”];输出:22
思路
逆波兰表达式是采用后缀表达式表示运算,那么每个二元运算符的操作对象都是在其前面值或表达式,这样我们可以采用栈表示需要被计算的操作数(表达式的结果也可以是之后运算符要计算的操作数),遍历数组若发现一个运算符,那栈顶的两个元素就是该二元运算符的操作数,将该两个元素出栈,将计算结果入栈,若是数值,则放入栈,作为后续运算符要计算的操作数。
复杂度
时间复杂度:
O ( n ) O(n) O(n)
空间复杂度:
O ( n ) O(n) O(n)
Code
class Solution:
def evalRPN(self, tokens: List[str]) -> int:
stack_val = []
for t in tokens:
if t not in ['+', '-', '*', '/']:
stack_val.append(int(t))
continue
x1 = stack_val.pop()
x2 = stack_val.pop()
if t=='+':
stack_val.append(x2 + x1)
elif t=='-':
stack_val.append(x2 - x1)
elif t=='*':
stack_val.append(x2 * x1)
elif t=='/':
stack_val.append(int(x2 / x1))
return stack_val.pop()