给你一个字符串数组 tokens
,表示一个根据 逆波兰表示法 表示的算术表达式。
请你计算该表达式。返回一个表示表达式值的整数。
注意:
- 有效的算符为
'+'
、'-'
、'*'
和'/'
。 - 每个操作数(运算对象)都可以是一个整数或者另一个表达式。
- 两个整数之间的除法总是 向零截断 。
- 表达式中不含除零运算。
- 输入是一个根据逆波兰表示法表示的算术表达式。
- 答案及所有中间计算结果可以用 32 位 整数表示。
示例 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
思路:大家应该对逆波兰表达式不陌生,实际上就是二叉树的后序遍历,在这里求值的时候呢,利用栈,每次遇到数字就入栈,遇到操作符就弹出2个元素进行相应的操作,思路不难,主要有几个需要注意的地方。
1.判断是否是数字,很多人会用isdigit来判断是不是数字,然后来入栈,但这个函数无法判断负数,比如例3的-11会被漏掉,那既然这样,我们就判断出现的这个字符是不是操作符,如果不是4个操作符中的任意一个,那一定就是数字了呗,然后将其入栈。
2.弹出的2个数字的运算顺序,在进行加或乘运算的时候,顺序不重要;但在进行减和除的时候,运算顺序就很重要了,一定是第二个弹出的元素去减或除第一个弹出的元素。
3.除法运算要向零截断,这个怎么实现呢,可以先进行浮点运算,然后将结果转成int型。先进行浮点运算,就能得到运算的浮点数,然后转成int,就能取到要的整数部分了。
代码(Python):
class Solution(object):
def evalRPN(self, tokens):
stack = []
for i,token in enumerate(tokens):
if token!='+' and token!='-' and token!='*' and token!='/': #数字直接入栈
stack.append(int(token))
else: #遇到操作符,弹出2个元素进行相应的操作
num1 = stack.pop()
num2 = stack.pop()
if token == '+':
stack.append(num1 + num2)
elif token == '-':
stack.append(num2 - num1)
elif token == '*':
stack.append(num1 * num2)
else:
stack.append(int(1.0 * num2 / num1))
return stack[0]