Leetcode 150. 逆波蘭表達式求值 棧

根據逆波蘭表示法,求表達式的值。

有效的運算符包括 +-*/ 。每個運算對象可以是整數,也可以是另一個逆波蘭表達式。

說明:

  • 整數除法只保留整數部分。
  • 給定逆波蘭表達式總是有效的。換句話說,表達式總會得出有效數值且不存在除數爲 0 的情況。

示例 1:

輸入: ["2", "1", "+", "3", "*"]
輸出: 9
解釋: ((2 + 1) * 3) = 9

示例 2:

輸入: ["4", "13", "5", "/", "+"]
輸出: 6
解釋: (4 + (13 / 5)) = 6

示例 3:

輸入: ["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

用棧來處理
atoi函數:把一個字符串變成一個整數,注意用“”。因爲默認是字符串。

class Solution {
public:
    int evalRPN(vector<string> &tokens) {
        if (tokens.size() == 1) 
        {
            return atoi(tokens[0].c_str());
        }
        stack<int> s;
        for (int i = 0; i < tokens.size(); ++i) 
        {
            if (tokens[i] != "+" && tokens[i] != "-" && tokens[i] != "*" && tokens[i] != "/") 
            {
                s.push(atoi(tokens[i].c_str()));
            }
            else 
            {
                int m = s.top();
                s.pop();
                int n = s.top();
                s.pop();
                if (tokens[i] == "+") s.push(n + m);
                if (tokens[i] == "-") s.push(n - m);
                if (tokens[i] == "*") s.push(n * m);
                if (tokens[i] == "/") s.push(n / m);
            }
        }
        return s.top();
    }
};

 

發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章