代码随想录算法训练营day9（栈与队列）

逆波兰表达式求值

• https://leetcode.cn/problems/evaluate-reverse-polish-notation/

思路

• 初始化一个栈
• 遍历整个tokens，遇到数字时将其压入到栈中，遇到运算符时，从栈中取两个数字进行运算，将运算完的结果压入到栈中，继续向后遍历
• 当遍历完后，留在栈中的元素就是最后的结果，return即可

class Solution:
    def evalRPN(self, tokens: List[str]) -> int:
        stack = []   # 存放的是整数
        arr = ['+','-','*','/']
        for i in tokens:
            if i not in arr:   # 说明是数字
                stack.append(int(i))
            else:
                num1 = stack.pop()
                num2 = stack.pop()
                if i == '+':
                    stack.append(num2 + num1)
                elif i == '-':
                    stack.append(num2 - num1)
                elif i == '*':
                    stack.append(num2 * num1)
                else:
                    stack.append(int(num2 / num1))
        return stack.pop()