227. 基本计算器 II

最新推荐文章于 2023-08-28 18:04:54 发布

Histranger_

最新推荐文章于 2023-08-28 18:04:54 发布

阅读量173

点赞数

本文链接：https://blog.csdn.net/Histranger_/article/details/114683473

版权

我一个滑铲专栏收录该内容

32 篇文章 0 订阅

订阅专栏

227. 基本计算器 II

LeetCode题目：227. 基本计算器 II

思路分析

题目大意：给你一个字符串表达式 s ，请你实现一个基本计算器来计算并返回它的值。整数除法仅保留整数部分。
有几个注意点：
- 运算符有+ - * /，其中/是整除。
- 没有负数。
- 数字的范围是整型。
刚开始想用栈，在写代码的过程中发现了一种自认为优美的解题方法。具体想法如下。
比如，计算s = "3+2*2"，显然应该优先计算2*2，一种通俗的思路是：先算乘除，再算加减 。如果我们算完乘除，那么s会变成s'=3+4，也就是我们算完所有的乘除以后，表达式中就只会有加减了，那么从左到右计算即可。
不妨假设表达式中只有加减法。那么，计算s的值是不是就很简单了？我们可以计ret存放当前的计算结果，ret_sign表示加号或者减号。
- 比如，s = "1+2-4"。如果当前位置在1处，那么，ret=0，ret_sign='+'（此时可以想象在s前面虚拟出来一个0+）；假设当前位置是2，那么，ret=1，ret_sign='+'；假设当前位置在4处，那么此时ret=3，ret_sign='-'。
- 显然，ret的初始值为0，ret_sign的初始值为+（不妨定义ret_sign=1时为+，ret_sign=0时为-）。
现在，我们再次做一个假设，如果表达式中只有乘除，问题是不是又一下子明朗了？不妨设tmp存放当前的计算结果，tmp_sign表示乘号或者除号。
- 举个栗子，比如，s = 2*2/4。如果当前位置在1处，那么，tmp=1，tmp_sign='*'（此时可以想象在s前面虚拟出来一个1*）；假设当前位置是2，那么，tmp=2，tmp_sign='*'；假设当前位置在4处，那么此时tmp=4，tmp_sign='/'。
- 显然，tmp的初始值为1，tmp_sign的初始值为*（不妨定义tmp_sign=1时为*，ret_sign=1时为/）。
但是，表达式中可能会有加减，也可能有乘除。但是，表达式s可以写成如下通用格式（暂时不考虑空格）:s = s1[+-]s2[+-]s3[+-]...si...sn，其中si是一个数字或者是只有乘除的表达式。
- 显然，对于s，我们可以认为只有加减法。对于任意一个si，只有乘除法。
我们认为s只有加减法的前提条件是s中得有加减号，不排除s = 1234这种s中没有加减号的情况。但是，我们可以使用一个奇淫技巧"哨兵"，我们让s=s+0，那么就能保证s中一定有加减号了。
有了上述思路，我么可以写出AC代码：

class Solution {
public:
    int calculate(string s) {
        s = s + "+0";
        int n = s.size();
        int pos = 0;
        int ret = 0;
        int tmp = 1;
        int ret_sign = 1; // 1 +; 0 -
        int tmp_sign = 1; // 1 *; 0 /
        while (pos < n){
            if (s[pos] == ' '){
                pos ++;
            }else if (s[pos] >= '0' && s[pos] <= '9'){
                int num = 0;
                while (pos < n && s[pos] >= '0' && s[pos] <= '9'){
                    num = num * 10 + (s[pos] - '0'); // 注意1
                    pos ++;
                }
                if (tmp_sign == 1){
                    tmp = tmp * num;
                }else{
                    tmp = tmp / num;
                }
            }else{
                if (s[pos] == '+'){
                    if (ret_sign == 1){
                        ret += tmp;
                    }else{
                        ret -= tmp;
                    }
                    ret_sign = 1;
                    tmp = 1;
                    tmp_sign = 1;
                }else if (s[pos] == '-'){
                    if (ret_sign == 1){
                        ret += tmp;
                    }else{
                        ret -= tmp;
                    }
                    ret_sign = 0;
                    tmp = 1;
                    tmp_sign = 1;
                }else if (s[pos] == '*'){
                    tmp_sign = 1;
                }else {
                    tmp_sign = 0;
                }
                pos ++;
            }
        }
        return ret;
    }
};

执行用时：4 ms, 在所有 C++ 提交中击败了99.44%的用户
内存消耗：8.1 MB, 在所有 C++ 提交中击败了80.06%的用户

代码略长，有很多代码可以简化，这里不再简化。比如：
```
if (ret_sign == 1){
    ret += tmp;
}else{
    ret -= tmp;
}
```
可以简化成一行：ret += (ret_sign ? -tmp : tmp)。
代码注释中的 注意1：注意，如果不加括号，可能在计算num * 10 + s[pos]时就会超出int范围。
复杂度分析：
- 时间复杂度:O(n)。一次遍历。
- 空间复杂度:O(1)。定义了可数个变量。