227. Basic Calculator II

题目:

Implement a basic calculator to evaluate a simple expression string.

The expression string contains only non-negative integers, +-*/ operators and empty spaces . The integer division should truncate toward zero.

Example 1:

Input: "3+2*2"
Output: 7

Example 2:

Input: " 3/2 "
Output: 1

Example 3:

Input: " 3+5 / 2 "
Output: 5

Note:

  • You may assume that the given expression is always valid.
  • Do not use the eval built-in library function.

 

思路:

这题是224的延申,同样也是用stack存数字,虽然看上去增加了乘和除,难了一点,但是它去掉了括号,因此medium难度。首先用stack存数字,这里存的是在符号前已经被操作完的数字,比如"3+2*2",遇到‘+’时,存的是{3},然后遇到‘+’时,存的是{3,2}。对于数字的处理,和以前一样,用一个int cur保存当前数字,每次遇到数字就乘十进位再加上遇到的数字。这里还会遇到空格,不过只要直接跳过即可。然后剩下的就是符号了,加减乘除,每次遇到符号,我们先处理手中的符号sign和数字cur。如果手里的sign是‘+’, 那么直接cur塞进stack,如果是‘-’,取相反数再塞进stack,因为里面存的是已经处理完的数字,我们最后的结果是整个stack的和(相当于把整个string看成多个小部分,那么负数只要取正数的反即可),‘*’则把当前数字乘以栈顶数字,‘/’同样的,栈顶数字除以当前数字。之后更新当前数字和当前符号,数字置为0,符号依据目前读到的符号即可。最后一小段的话出了循环还是要处理一下手中的符号和数字,这一段写的有点难看,理论上可以直接放入循环内的,比如当i==s.size()-1的时候,进行运算等等,不过不影响结果,就是这种可以改造的重复确实很丑。最后的最后把stack中的所有数相加即是结果。

 

代码:

class Solution {
public:
    int calculate(string s) {
        stack<int> nums;
        int cur=0;
        char sign='+';
        for(auto c:s)
        {
            if(isdigit(c))
                cur=cur*10+(c-'0');
            else if(c==' ')
                continue;
            else
            {
                if(sign=='+')
                    nums.push(cur);
                else if(sign=='-')
                    nums.push(-cur);
                else if(sign=='*')
                    nums.top()=nums.top()*cur;
                else if(sign=='/')
                    nums.top()=nums.top()/cur;
                sign=c;
                cur=0;
            }
        }
        if(sign=='+')
            nums.push(cur);
        else if(sign=='-')
            nums.push(-cur);
        else if(sign=='*')
            nums.top()=nums.top()*cur;
        else if(sign=='/')
            nums.top()=nums.top()/cur;
        int ans=0;
        while(nums.size())
        {
            ans+=nums.top();
            nums.pop();
        }
        return ans; 
    }
};

 

已标记关键词 清除标记
©️2020 CSDN 皮肤主题: 数字20 设计师:CSDN官方博客 返回首页