LeetCode: Evaluate Reverse Polish Notation [150]

標籤：leetcode   演算法   面試   

【題目】

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 str2int(string token){        int num=0;        int start=0;        int isNev = 1;        //判斷符號        if(token[0]=='-'){isNev=-1; start++;}        else if(token[0]=='+')start++;        for(int i=start; i<token.length(); i++){            num=10*num+(token[i]-'0');        }        return num*isNev;    }        bool isOp(string token){        if(token=="/" || token=="+" || token=="-" || token=="*")            return true;    }        int evalRPN(vector<string> &tokens) {        stack<int> st;        for(int i=0; i<tokens.size(); i++){            if(isOp(tokens[i])){                //如果是運算子，則計算棧頂元素，然後將結果入棧                int val1 = st.top(); st.pop();                int val2 = st.top(); st.pop();                int res = 0;                if(tokens[i]=="+")  res = val2 + val1;                else if(tokens[i]=="-")  res = val2 - val1;                else if(tokens[i]=="*")  res = val2 * val1;                else if(tokens[i]=="/")  res = val2 / val1;                //計算結果入棧                st.push(res);            }            else{                //數字入棧                st.push(str2int(tokens[i]));            }        }        return st.top();    }};