227. 基本计算器 II

在这里插入图片描述
一般要处理符号的,就直接想到栈,又是一道细节题。
我采用的方法是先遍历一遍,将*/,然后将栈中的元素依次弹出相加。
来一个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;
    }
};
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值