leetcode#150 逆波兰表达式求值

leetcode#150 逆波兰表达式求值

题目：

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

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

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

示例：

输入：tokens = ["2","1","+","3","*"]
输出：9
解释：该算式转化为常见的中缀算术表达式为：((2 + 1) * 3) = 9

思路：

可以想到对于逆波兰式的每个运算符，都是前两个数字进行运算然后放回去，因此我们可以使用栈进行模拟。

代码:

class Solution
{
public:
    int evalRPN(vector<string> &tokens)
    {
        stack<int> st;
        int n = tokens.size();
        for (int i = 0; i < n; ++i)
        {
            if (tokens[i] == "*")
            {
                int tmp = st.top();
                st.pop();
                st.top() *= tmp;
            }
            else if (tokens[i] == "-")
            {
                int tmp = st.top();
                st.pop();
                st.top() -= tmp;
            }
            else if (tokens[i] == "+")
            {
                int tmp = st.top();
                st.pop();
                st.top() += tmp;
            }
            else if (tokens[i] == "/")
            {
                int tmp = st.top();
                st.pop();
                st.top() /= tmp;
            }
            else
            {
                int sum = 0, j = 0;
                if (tokens[i][0] == '-' || tokens[i][0] == '+')
                    j = 1;
                for (; j < tokens[i].length(); ++j)
                    sum = sum * 10 + tokens[i][j] - '0';
                if (tokens[i][0] == '-')
                    sum = -sum;
                st.push(sum);
            }
        }
        return st.top();
    }
};