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