LeetCode 150. 逆波兰表达式求值

题目描述

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

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

说明：

• 整数除法只保留整数部分。
• 给定逆波兰表达式总是有效的。换句话说，表达式总会得出有效数值且不存在除数为 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) {
        stack<int> s;
        for(int i = 0; i < tokens.size(); i++){
            if(tokens[i] != "+" && tokens[i] != "-" && tokens[i] != "*" && tokens[i] != "/")
                s.push(stoi(tokens[i]));
            else{
                int temp1 = s.top();
                s.pop();
                int temp2 = s.top();
                s.pop();
                if(tokens[i] == "+")
                    s.push(temp2 + temp1);
                if(tokens[i] == "-")
                    s.push(temp2 - temp1);
                if(tokens[i] == "*")
                    s.push(temp2 * temp1);
                if(tokens[i] == "/")
                    s.push(temp2 / temp1);
            }
        }
        return s.top();
    }
};