栈和队列-逆波兰表达式求值

Problem: 150. 逆波兰表达式求值

文章目录

  • 题目
  • 思路
  • 复杂度
  • Code

题目

输入一个根据逆波兰表达式表示的四则运算表达式,输出计算结果。注意除法是向零取整。
示例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()

你可能感兴趣的:(数据结构和算法,leetcode,算法)