一般要处理符号的,就直接想到栈,又是一道细节题。
我采用的方法是先遍历一遍,将*/,然后将栈中的元素依次弹出相加。
来一个char记录当前数字的符号,比如3-5,那么入栈的就是3和-5,所以prejection初始化为‘+’,遍历字符串,当遍历到空格时直接跳过,遍历到数字时依次计算。
当遍历到字符时,如果prejection是+,就将最新计算出的cur入栈,如果是-,就将-cur入栈。如果是*,那么证明是优先计算的,要将cur和栈顶元素相乘,更新栈顶元素。如果是/,就让栈顶元素/cur。将cur重置为0,并记录当前符号是什么,因为当前符号是下一个出现的数字的符号。
最后stack中存的都是数字。
重点来了,如果是2+3这种,遍历完3后,后面没有符号的怎么办,那么3就入不了栈了,所以在一开始,s最后就加上一个特殊符号,这样可以保证最后一个数字正常入栈
class Solution {
public:
int calculate(string s) {
//先算*/,再算加减
//用一个字符记录当前数字的前一个符号是什么
//如果是2*2,那么最后一个2后面没有特殊符号了,所以最后一个*2不会被计算进去
//所以最后给s最后加上一个特殊符号
s += "$";
char prejection = '+';
int res = 0, cur = 0;
stack<int> sta;
for(int i = 0; i < s.length(); ++i){
if(s[i] == ' ') continue;
if(isdigit(s[i])){
cur = cur * 10 +(s[i]-'0');
}
if(!isdigit(s[i])){
if(prejection == '+') sta.push(cur);
else if(prejection == '-') sta.push(-cur);
else if(prejection == '*') sta.top() = sta.top()*cur;
else sta.top() = sta.top()/cur;
cur = 0, prejection = s[i];
}
}
while(!sta.empty()){
res += sta.top();
sta.pop();
}
return res;
}
};