[leetcode] 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.

You may assume that the given expression is always valid.

Some examples:

"3+2*2" = 7
" 3/2 " = 1
" 3+5 / 2 " = 5

Note:Do not use the eval built-in library function.

这道题是完成一个包含加减乘除的计算器，题目难度为Medium。

我们知道，乘除的优先级比加减高，因此我们需要优先计算乘除的结果。遍历整个表达式，遇到‘+’或‘-’时表明之前的乘除（或没有）已经结束，此时可以对前面的乘除结果（没有乘除即是一个单独数字）进行计算了。在遇到‘*’或‘/’时不能立即将之前出现的数字进行计算，而要将乘除的结果缓存起来等待‘+’或‘-’的出现。具体代码：

class Solution {
public:
    int calculate(string s) {
        int num = 0, rst = 0;
        int lastNum = 0;
        int markOp = 1, lastOp = 1; //1:+, -1:-, 2:*, 3:/
        
        s += "+";
        for(auto ch:s) {
            if(ch >= '0' && ch <= '9') {
                num = num*10+ch-'0';
            }
            else if(ch == ' ') {
                continue;
            }
            else {
                if(lastOp < 2) lastNum = num;
                else if(lastOp == 2) lastNum *= num;
                else lastNum /= num;
                num = 0;
                
                if(ch == '+' || ch == '-') {
                    rst += markOp * lastNum;
                    markOp = (ch == '+') ? 1 : -1;
                    lastOp = markOp;
                    lastNum = 0;
                }
                else {
                    lastOp = (ch == '*') ? 2 : 3;
                }
            }
        }
        
        return rst;
    }
};

数字的处理就不再详述了，在遇到运算符时，根据上一个运算符的类型进行不同的处理，如果上一个运算符是‘+’或‘-’，把当前数字缓存起来等候处理（如果当前运算符是‘*’或‘/’，缓存起来的数据就是乘除运算的初始数值）；如果上一个运算符是‘*’或‘/’，说明乘除运算已经开始，用缓存的乘除结果对当前数值进行乘除作为新的乘除结果缓存起来。

如果当前运算符是’+‘或’-‘，表明之前的乘除（或没有）已经结束，可以将其结果运算到最终结果中了，用我们记录下来的上一个加减运算的运算符（markOp）对乘除结果（没有乘除运算就是一个单独的数字）进行计算，同时修改上一个运算符标志以及上一个加减运算符标志，并将记录的乘除运算缓存清零。如果当前运算符是’*‘或’/‘，修改上一个运算符标志即可。听起来可能很绕，可以直接查看代码再和解释比对来理解。