旺崽的博客

要么天赋异禀,要么天道酬勤

0%

Leetcode 150 逆波兰表达式求值

题目链接

根据逆波兰表示法,求表达式的值。

有效的算符包括 +、-、*、/ 。每个运算对象可以是整数,也可以是另一个逆波兰表达式。

说明:

1
2
整数除法只保留整数部分。
给定逆波兰表达式总是有效的。换句话说,表达式总会得出有效数值且不存在除数为 0 的情况。

示例 1:

1
2
3
输入:tokens = ["2","1","+","3","*"]
输出:9
解释:该算式转化为常见的中缀算术表达式为:((2 + 1) * 3) = 9

示例 2:

1
2
3
输入:tokens = ["4","13","5","/","+"]
输出:6
解释:该算式转化为常见的中缀算术表达式为:(4 + (13 / 5)) = 6

示例 3:

1
2
3
4
5
6
7
8
9
10
11
输入: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

简单栈操作,用一个数字栈存所有数字,每碰到一个运算符,就从栈中取两个数计算出一个结果再存到栈里,最后返回栈顶即可,AC代码如下:

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
class Solution:
def evalRPN(self, tokens: List[str]) -> int:
num = []
for i in tokens:
if i != '+' and i != '-' and i != '*' and i != '/':
num.append(int(i))
else:
a = num[-1]
num.pop()
b = num[-1]
num.pop()
if i == '+':
num.append(a + b)
elif i == '-':
num.append(b - a)
elif i == '/':
num.append(int(b / a))
elif i == '*':
num.append(a * b)
return num[-1]