150. 逆波兰表达式求值
【中等】
给你一个字符串数组 tokens
,表示一个根据 逆波兰表示法 表示的算术表达式。
请你计算该表达式。返回一个表示表达式值的整数。
注意:
- 有效的算符为
'+'
、'-'
、'*'
和'/'
。 - 每个操作数(运算对象)都可以是一个整数或者另一个表达式。
- 两个整数之间的除法总是 向零截断 。
- 表达式中不含除零运算。
- 输入是一个根据逆波兰表示法表示的算术表达式。
- 答案及所有中间计算结果可以用 32 位 整数表示。
后缀表达式转中缀表达式:
- 遇到操作数,入栈。
- 遇到运算符,弹出栈顶的两个操作数,运算后将结果压入栈。
注意点:
- 对于除法运算来说,两个操作数的先后关系至关重要,要清楚栈顶弹出的第一个数字是算数式里的num2。
- python3的除法规则是浮点除法,int()转换后刚好是向零取整。
- 判断是否为数字比较复杂,应该先判断是否是运算符
class Solution:
def evalRPN(self, tokens: List[str]) -> int:
stack = []
for i in tokens:
# 判断是否是数字有些困难--先判断是否是运算符
if stack and i in "+-*/":
num2 = stack.pop() #注意栈顶元素是第二个操作数,对除法有影响
num1 = stack.pop()
if i =="+":
stack.append(num1+num2)
elif i =="-":
stack.append(num1-num2)
elif i =="*":
stack.append(num1*num2)
elif i =="/":
# python的除法是浮点除法,且向下取整,题目要求向0取整
# int() 函数会去掉小数部分,而不是进行四舍五入。
# 正数处理:相当于向零取整
# 负数处理:对于负数,int() 依然会去掉小数部分,结果会向零的方向取整。
stack.append(int(num1/num2))
else:
stack.append(int(i)) # 是操作数 加入栈
return stack.pop() #最后结果弹出
- 时间复杂度: O(n)
- 空间复杂度: O(n)