【20190715】【每天一道算法题】逆波兰表达式求值（栈）

问题

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

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

说明：

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

示例 1：

输入: ["2", "1", "+", "3", "*"]，输出: 9

解释: ((2 + 1) * 3) = 9

示例 2：

输入: ["4", "13", "5", "/", "+"]，输出: 6

解释: (4 + (13 / 5)) = 6

示例 3：

输入: ["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

---

思路及解答

# 方法：栈操作（Python中没有内建栈操作，需要用列表的 .append() 和 .pop() 来模拟栈操作）
# 遍历列表，如果没有遇到操作符，依次进栈。若遇到操作符，从栈顶弹出两个，做相应运算（注意弹出顺序和操作顺序的对应），并将操作结果压入栈，再等待后续操作，直到列表元素全部参与运算，最终弹出栈顶元素。
# 时间复杂度：O(n)，因为只进行了一次遍历。进栈 即.append() 和出栈 即.pop() 的时间复杂度都是O(1)。
class Solution:
    def evalRPN(self, tokens: List[str]) -> int:
        stack = []
        for item in tokens:
            if (item != "+") and (item != "-") and (item != "*") and (item != "/"):
                stack.append(item)   # 进栈操作
            else:
                top1 = int(stack.pop())   # 弹出两个栈顶元素
                top2 = int(stack.pop())
                if item == "+":
                    stack.append(top2 + top1)
                elif item == "-":
                    stack.append(top2 - top1)
                elif item == "*":
                    stack.append(top2 * top1)
                else:                # 除法运算有点复杂（要考虑一正一负和元素为零的情况）
                    if (top1 > 0 and top2 > 0) or (top1 < 0 and top2 < 0):
                        stack.append(top2 // top1)
                    elif top1 * top2 < 0:
                        stack.append(-(-top2 // top1))
                    else:
                        stack.append(0)
        return int(stack.pop())    # 列表元素都是字符类型(str)，所以要转换为整数输出。

---

知识点

1. 时间复杂度的问题

进栈和出栈操作时间复杂度都是 O(1)。

其他的 Python 操作的时间复杂度见参考：时间复杂度（大O表示法）以及使用python实现栈

2. Python报错 TypeError: bad operand type for unary +: 'str'  通常是 缩进 的问题。

3. Reverse Polish Notation（逆波兰表达式）

操作对象在操作符之前，操作符只有 "+", "-", "*", "/"，而没有括号，所以操作起来很简单。

通常的二元操作符大多都位于两个操作数之间，所以称作中缀表达式；而逆波兰表达式所有的操作符位于操作数的后面，所以也叫做后缀表达式。

参考：逆波兰表达式百度百科