leetcode 227. 基本计算器 II

一、题目

• 1 <= s.length <= 3 * 105
• s 由整数和算符 (’+’, ‘-’, ‘*’, ‘/’) 组成，中间由一些空格隔开
• s 表示一个 有效表达式
• 表达式中的所有整数都是非负整数，且在范围 [0, 231 - 1] 内
• 题目数据保证答案是一个 32-bit 整数

输入：s = “3+2*2”
输出：7

二、代码

class Solution {
public:
    bool isdigit(char ch){
        return ch>='0'&&ch<='9';
    }

    int calculate(string s) {
        int len=s.length(),num=0;
        vector<int> stk;
        char ch,prevSign='+';
        for(int i=0;i<len;++i){
            ch=s[i];
            if(isdigit(ch)){
                num=num*10+(ch-'0');
            }
            if(ch!=' '&&!isdigit(ch)||i==len-1){
                switch(prevSign){
                    case '+': stk.push_back(num);break;
                    case '-': stk.push_back(-num);break;
                    case '*': stk.back()*=num; break;
                    case '/': stk.back()/=num; break;
                }
                prevSign=ch;
                num=0;
            }
            
        }
        return accumulate(stk.begin(),stk.end(),0);
    }
};

解题思路：

1. 我一开始用的双栈，但发现时间太久了，后来发现，有括号的计算器才需要用双栈。
2. 无括号的计算器，只用数字栈即可。需要保存一个之前符号prevSign，刚开始prevSign=’+’，之后每读取一个新操作符/到达字符串结尾，若prevSign为加减，则根据符号，将数字num入栈；若为乘除，则直接将栈顶元素和num做乘除。
3. 最后栈内元素相加即可，这里的栈用的是vector，用了一个accumulate(vector.begin(),vector.end(),0)可以将vector所有元素相加。