LeetCode -- Evaluate Reverse Polish Notation

題目如下：

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

-----------------------------------------------------------------------------------

這個題目難度也不大，主要是利用棧後進先出的性質，每次遇到數字就進棧，遇到運算符就拿出兩個數字用這個運算符計算，然後將計算的結果保存到棧中。

如果表達式沒有錯誤的話，最後棧頂的元素就是計算的結果。

class Solution {
public:
    int evalRPN(vector<string> &tokens) {
        stack<int> stk;
        for(vector<string>::iterator it = tokens.begin(); it != tokens.end(); ++it)
        {
            if(isOper(*it) ){
                int b = stk.top(); stk.pop();
                int a = stk.top(); stk.pop();
                stk.push(rst(a, b, *it));
            }else{
                stk.push(numb(*it));
            }
        }
        return stk.top();
    }
    int numb(string s)
    {
        int ret = 0, flag = 1;
        size_t i = 0;
        if(s[i] == '-'){
            flag = -1;
            ++i;
        }else if(s[i] == '+'){
            ++i;
        }
        for(; i < s.length(); ++i)
        {
            ret = ret*10 + (s[i] - '0');
        }
        return ret*flag;
    }
    int rst(int a, int b, string op)
    {
        if(op == "+")
            return a+b;
        else if(op == "-")
            return a-b;
        else if(op == "*")
            return a*b;
        else if(op == "/")
            return a/b;
    }
    bool isOper(string s){
        return (s == "+" || s == "-" || s == "*" || s == "/");
    }
};

代碼中，用isOper判斷元素是運算符還是數字，如果是運算符就從棧中取兩個數字，用rst來做計算，如果是數字的話就轉化位數字直接進棧。

在進行數字轉化的時候，考慮了數字符號爲正和爲負兩種情況，不知道LeetCode的測試用例有沒有考慮到這個情況。

代碼的長度還是比較長的，可以進一步優化，不過這麼寫，更容易看懂吧