LeetCode 题解

积跬步，至千里。

逆波兰表达式求值

分类：LeetCode 标签： 栈 发布于：2020-07-07 ID：150

题目描述

来源于 https://leetcode-cn.com/

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

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

说明：

  • 整数除法只保留整数部分。
  • 给定逆波兰表达式总是有效的。换句话说，表达式总会得出有效数值且不存在除数为 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

解法：

class Solution {
  public:
    int evalRPN(vector<string> &tokens) {
        if (tokens.empty()) return 0;
        stack<int> stk;
        for (auto token : tokens) {
            if (is_num(token)) {
                stk.push(strtol(token.c_str(), nullptr, 10));
            } else {
                int rhs = stk.top();
                stk.pop();
                int lhs = stk.top();
                stk.pop();
                char ch = token.back();
                switch (ch) {
                case '+':
                    stk.push(lhs + rhs);
                    break;
                case '-':
                    stk.push(lhs - rhs);
                    break;
                case '*':
                    stk.push(lhs * rhs);
                    break;
                case '/':
                    stk.push(lhs / rhs);
                    break;
                default:
                    break;
                }
            }
        }
        return stk.top();
    }

    bool is_num(const string &token) {
        char ch = token.back();
        return ch >= '0' && ch <= '9';
    }
};