Bootstrap

LeetCode 227. 基本计算器 II

更多题解尽在 https://sugar.matrixlab.dev/algorithm 每日更新。
组队打卡,更多解法等你一起来参与哦!

LeetCode 227. 基本计算器 II,难度中等

解题思路:创建一个栈用于保存操作数。遍历字符串,如果当前字符是“ ± 运算符”,则压入栈中;如果当前字符是“ */ 运算符”,则从栈中取数计算后重新压入栈中。最后将栈中的数据累加。

为什么运算符是后更新?
这是因为我们需要在处理当前数字和前一个运算符之后再更新运算符,以确保当前数字被正确处理。

class Solution {
    public int calculate(String s) {
        if (s.isEmpty()) return 0;
        Stack<Integer> stack = new Stack<>();
        int result = 0;
        // 当前数字
        int num = 0;
        char sign = '+';
        for (int i = 0; i < s.length(); ++i) {
            char c = s.charAt(i);
            if (Character.isDigit(c)) {
                // 判断并计算出数字
                num = num * 10 + (c - '0');
            }

            if ((!Character.isDigit(c) && c != ' ') || i == s.length() - 1) {
                switch (sign) {
                    case '+':
                        // 将当前数字压入栈
                        stack.push(num);
                        break;
                    case '-':
                        // 将当前数字的相反数压入栈
                        stack.push(-num);
                        break;
                    case '*':
                        // 弹出栈顶元素,与当前数字相乘后压入栈
                        stack.push(stack.pop() * num);
                        break;
                    case '/':
                        // 弹出栈顶元素,与当前数字相除后压入栈
                        stack.push(stack.pop() / num);
                        break;
                }
                // 更新运算符
                sign = c;
                num = 0;
            }
        }

        // 将栈内数据累加
        for (int i : stack) {
            result += i;
        }
        return result;
    }
}
;