【LeetCode】227。基本電卓II（C ++）

---

件名のソース：https：//leetcode-cn.com/problems/basic-calculator-ii/

タイトル説明

文字列式を与えてください。その値を計算して返すための基本的な計算機を実装してください。

整数除算は整数部分のみを保持します。

例1：

入力：s = "3 + 2 2"
出力：7
例2：

入力：s = "3/2"
出力：1
例3：

入力：s = "3 + 5/2 "
出力：5

ヒント：

1 <= s.length <= 3 * 105
sは、整数と演算子（ '+'、 '-'、 ' '、 '/'）で構成され、いくつかのスペースで区切られます。s
は、
式の有効な式を意味します。すべての整数inは非負の整数であり、[0、231-1]の範囲にあります。
質問データは、回答が32ビット整数であることを保証します。

一般的なアイデア

• 動的データスタックと演算子スタックを維持するためだけに、昨日の問題ははるかに単純であり、データスタックが残りの文字数であり、op'- 'のスタックがこの時点でSignである必要がある場合にのみ注意を払う必要があります。否定

スタック

class Solution {
    
    
public:
    int calculate(string s) {
    
    
        stack<char> op;
        stack<int> nums;
        int len = s.size();
        for (int i = 0 ; i < len ;){
    
    
            if(s[i] == ' ') ++i;
            if(s[i] == '+' || s[i] == '-' || s[i] == '*' || s[i] == '/'){
    
    
                op.push(s[i]);
                ++i;
            }
            if(isdigit(s[i])){
    
    
                int num = 0;
                while(isdigit(s[i])){
    
    
                    num =  num * 10 + int(s[i] - '0');
                    ++i;
                }
                if(!op.empty() && op.top() == '-'){
    
    
                   num *= -1;
                   op.pop();
                   op.push('+');
                }
                nums.push(num);
                if(!op.empty() && (op.top() == '*' || op.top() == '/')){
    
    
                    ReallyCalculate(op.top(), nums);
                    op.pop();
                }
            }
        }
        // 当数据栈中只剩一个操作数时出来
        while(nums.size() != 1){
    
    
            ReallyCalculate(op.top(), nums);
            op.pop();
        }
        int ans = nums.top();
        if(nums.size() == 1 && !op.empty() && op.top() == '-')
            ans *= -1;
        return ans;
    }
    void ReallyCalculate(char op, stack<int>& nums){
    
    
        int secondNum = nums.top();
        nums.pop();
        int firstNum = nums.top();
        nums.pop();
        switch(op){
    
    
            case '*':
                nums.push(firstNum * secondNum);
                break;
            case '/':
                nums.push(firstNum / secondNum);
                break;
            case '+':
                nums.push(firstNum + secondNum);
                break;
            case '-':
                nums.push(firstNum - secondNum);
                break;
        }
    }
};

• 公式ソリューション

typedef long long ll;
class Solution {
    
    
public:
    int calculate(string s) {
    
    
        vector<int> nums;
        int len = s.size();
        int num = 0;
        char op = '+';
        for(int i = 0 ; i < len ; ++i){
    
    
            if(s[i] >= '0' && s[i] <= '9')
                num = num * 10 + int(s[i] - '0');
            if(!isdigit(s[i]) && s[i] != ' '|| i == len - 1){
    
    
                switch (op){
    
    
                    case '+':
                        nums.push_back(num);
                        break;
                    case '-':
                        nums.push_back(-num);
                        break;
                    case '*':
                        nums.back() *= num;
                        break;
                    case '/':
                        nums.back() /= num;
                        break;
                }
                num = 0;
                op = s[i];
            }
        }

        return accumulate(nums.begin(), nums.end(), 0);
    }
};

複雑さの分析

• 時間計算量：O（n）。nは配列の長さです
• スペースの複雑さ：O（n）。nは配列の長さです