https://leetcode-cn.com/problems/basic-calculator/
アイデア:なんて退屈な問題...細部に注意を払うだけで、直接再帰することができます。
class Solution {
public:
using pr=pair<int,int>;
pr dfs(const string &s,int idx)
{
int n=s.size(),sum=0,tmp=0,sign=1;
while(idx<n)
{
if(s[idx]=='(')
{
pr p=dfs(s,idx+1);
sum+=sign*p.first;
idx=p.second;
tmp=0,sign=1;
}
else if(s[idx]==')')
break;
else if(s[idx]==' ')
;
else if(s[idx]>='0'&&s[idx]<='9')
tmp=tmp*10+(s[idx]-'0');
else
{
sum+=sign*tmp;
tmp=0;
sign=s[idx]=='+'?1:-1;
}
++idx;
}
sum+=sign*tmp;
return pr(sum,idx);
}
int calculate(string s) {
return dfs(s,0).first;
}
};
または、スタックを使用します。現在のシンボルを格納するという考え方です。正の記号は1、負の記号は-1です。正確には、角かっこの前に記号を保持するため、最終的な正しい記号を直接計算できます。角かっこ内(かっこを入れるのと同じです)。
class Solution {
public:
int calculate(string s) {
stack<int> signs;
int ans=0,sign=1,n=s.size(),tmp=0,i=0;
signs.push(1);
while(i<n)
{
if(s[i]==' ')
;
else if(s[i]=='(')
signs.push(sign);
else if(s[i]==')')
signs.pop();
else if(s[i]=='+')
sign=signs.top();
else if(s[i]=='-')
sign=-signs.top();
else
{
tmp=0;
while(i<n&&s[i]>='0'&&s[i]<='9')
tmp=tmp*10+(s[i]-'0'),++i;
ans+=sign*tmp;
--i;
}
++i;
}
return ans;
}
};