【LeetCode】150. 逆波兰表达式求值

题目链接：https://leetcode-cn.com/problems/evaluate-reverse-polish-notation/description/

题目描述

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

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

说明：

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

示例1

输入: [“2”, “1”, “+”, “3”, “*”]
输出: 9
解释: ((2 + 1) * 3) = 9

示例2

输入: [“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.size()==0) return 0;
        if (tokens.size()==1) return stoi(tokens[0]);
        
        stack<int> st;
        for (int i=0;i<tokens.size();i++){
            if (tokens[i]=="+" || tokens[i]=="-" || tokens[i]=="*"  || tokens[i]=="/"){
                int num1=st.top();st.pop();
                int num2=st.top();st.pop();
                if (tokens[i]=="+") st.push(num2+num1);
                if (tokens[i]=="-") st.push(num2-num1);
                if (tokens[i]=="*") st.push(num2*num1);
                if (tokens[i]=="/") st.push(num2/num1);
            }            
            else
                st.push(stoi(tokens[i])); //stoi函数用于string转int
                
        }
        return st.top();
    }
};