这道题最traverse的解法就是,对于位于i的符号,计算在他前面的结果。如果遇到'(',那么就保护现场,刷新临时变量开始下一个计算。遇见')'说明当前小节结束,merge当前的临时变量和stack最顶端保留的计算符号和结果。
代码条路清晰,结构冗余。
int calculate(string s) {
stack<int> memo;
int res = 0;
int sign = 1;
int num=0;
for(int i=0;i<s.size();++i){
if(isdigit(s[i])){
num=num*10+s[i]-'0';
continue;
}
else if(s[i]=='+'){
res+=sign*num;
sign=1;
num=0;
}
else if(s[i]=='-'){
res+=sign*num;
sign=-1;
num=0;
}
else if(s[i]=='('){
memo.push(res);
memo.push(sign);
sign=1;
res=0;
}
else if(s[i]==')'){
res+=sign*num;
num=0;
res*=memo.top();
memo.pop();
res+=memo.top();
memo.pop();
}
}
if(num!=0)
res+=sign*num;
return res;
}
网上普遍的解法更短小精悍,不过稍微有一点点绕。 每次遇到一个运算符或者开始一个新的运算集团,都要想stack里面push一个运算符号。如果括号之前的符号是减号,就把括号里的所有符号都变成相反的。这样就可以平铺直叙的从头到尾完成整个运算。每进行完一次运算之后,舍弃当前的运算符号。‘)’ 同理,因为在遇见'('的时候我们在第一个括号内的数字前加了想象的‘+’。
这个解法值得注意的一点就是memo的初始化。memo[0]表示整个表达式的符号,memo[1]是加在第一个数之前的正好。比如在1+1这种情况的时候,运算第一个1之后把memo最后面的加号pop出来。在遇见‘+’的时候,程序会去看之前有没有运算符号对当前运算符号产生影响/*比如-(3+1)这种情况*/。若memo为空,那么memo.back()的值会返回0,所以整个表达式返回1.
int calculate(string s) {
vector<int> memo(1,1);
int res=0;
for(int i=0;i<s.size();++i){
char c=s[i];
if(c>='0'&&c<='9'){
int num=0;
while(i<s.size()&&s[i]>='0'&&s[i]<='9'){
num=num*10 + s[i++]-'0';
}
res+=memo.back()*num;
memo.pop_back();
--i;
}
else if(c==')') memo.pop_back();
else if(c=='+'||c=='-'||c=='(') memo.push_back(memo.back()*(c=='-'?-1:1));
//cout<<res<<endl;
}
return res;
}