leetcode-Basic Calculator-224

Posted 2020-08-06 0_summer

输入一个字符串，包含数字或者+,-,(,)，求这个表达式的结果

1.没有括号时顺序执行

2.有括号时先计算括号内的表达式

这里用一个栈来保存遇到括号(之前的表达式的结果，也就是说没遇到括号之前，按顺序正常执行设结果为x，遇到括号之后要先计算括号内的表达式，设结果为y，然后再把y拿出来和x顺序执行。因此计算括号内的表达式时要先保存之前计算的x的值，为什么要用栈呢，因为括号可以嵌套，如a-(b-(c+d))，遇到第一个括号时，要保存a，然后进入括号，又遇到了括号要保存b，然后进入第二个括号得到c+d之后返回和b进行计算，再返回和a进行计算，这就是栈的后进先出。当然还要把符号也压栈

这里对+和-直接用一个标志flag=1和flag=-1来表示，这样计算a-b时可以直接a+b*flag

 1 class Solution {
 2 public:
 3     int calculate(string s) {
 4         int len=s.size();
 5         if(len==0) return 0;
 6         stack<int> st;
 7         int ans=0;
 8         int i=0;
 9         int flag=1;
10         while(i<len){
11             if(s[i]==‘ ‘){
12                 i++;
13                 continue;
14             }
15             if(isdigit(s[i])){
16                 int num=0;
17                 while(i<len&&isdigit(s[i])){
18                     num=num*10+s[i]-‘0‘;
19                     i++;
20                 }
21                 ans+=num*flag;
22                 continue;
23             }
24             if(s[i]==‘+‘) flag=1;
25             else if(s[i]==‘-‘) flag=-1;
26             else if(s[i]==‘(‘){
27                 st.push(ans);
28                 ans=0;
29                 st.push(flag);
30                 flag=1;
31             }
32             else{
33                 int a=st.top();
34                 st.pop();
35                 int b=st.top();
36                 st.pop();
37                 ans=b+a*ans;
38             }
39             i++;
40         }
41         return ans;
42     }
43 };