【leetcode】Evaluate Reverse Polish Notation（middle）

Evaluate the value of an arithmetic expression in Reverse Polish Notation.

Valid operators are +, -, *, /. Each operand may be an integer or another expression.

Some examples:

  ["2", "1", "+", "3", "*"] -> ((2 + 1) * 3) -> 9
  ["4", "13", "5", "/", "+"] -> (4 + (13 / 5)) -> 6

思路：

题目的意思是表达式总是先给出两个操作数，然后再给出对应的操作符。当给出一个这样的表达式后，求出表达式对应的值。

难点是后面操作符的操作数也是表达式。解决方法是用一个栈nums来保存所有没有用过的操作数，当遇到操作符时，栈最上面的两个数字就是操作符对应的操作数。最先遇到的操作符他的操作数一定没有嵌套表达式，所以可以得到值，当运算得到新的数字后，压入栈，作为下一个操作符的操作数。

我的代码：数字与字符串转换的时候比较丑。

int evalRPN2(vector<string>& tokens)
    {
        if(tokens.size() == 1)
            return atoi(tokens[0].c_str());
        vector<string> nums;
        for(int i = 0; i < tokens.size(); i++)
        {
            if(tokens[i] == "+" || tokens[i] == "-" || tokens[i] == "*" || tokens[i] == "/")
            {
                string sPreNum, sPostNum, sOp;
                int preNum, postNum, tmp;
                sPostNum = nums.back();
                nums.pop_back();
                sPreNum = nums.back();
                nums.pop_back();
                preNum = atoi(sPreNum.c_str());
                postNum = atoi(sPostNum.c_str());
                sOp = tokens[i];
                switch(sOp[0])
                {
                case '+':
                    tmp = preNum + postNum; break;
                case '-':
                    tmp = preNum - postNum; break;
                case '*':
                    tmp = preNum * postNum; break;
                case '/':
                    tmp = preNum / postNum; break;
                default:
                    break;
                }
                char ctmp[20];
                sprintf(ctmp, "%d", tmp);
                nums.push_back(ctmp);
            }
            else
            {
                nums.push_back(tokens[i]);
            }
        }
        return atoi(nums[0].c_str());
    }

大神用递归的答案，相对短一些，但是逻辑不好理解。

int eval_expression(vector<string>& tokens, int& pt)
{
    string s = tokens[pt];

    if(s == "+" || s == "-" || s == "*" || s== "/") // tokens[r] is an operator
    {
        pt--;
        int v2 = eval_expression(tokens, pt);
        pt--;
        int v1 = eval_expression(tokens, pt);
        if(s == "+")
            return v1 + v2;
        else if(s == "-")
            return v1 - v2;
        else if(s == "*")
            return v1 * v2;
        else 
            return v1 / v2;
    }
    else // tokens[r] is a number
    {
        return atoi(s.c_str());
    }
}

int evalRPN(vector<string> &tokens) {
    int pt = tokens.size()-1; 
    return eval_expression(tokens, pt);
}