LeetCode-227. 基本计算器 II-Java-medium

题目链接

法一（栈）

    /**
     * 法一（栈）
     * 思路：
     * （1）一般需要符号栈、数据栈两个栈
     * （2）这里只用了一个数据栈，符号栈用一个变量sign代替了，只存储上一个符号
     * 注意
     * （1）表达式中没有括号
     * （2）加减乘除空格的ASCII码都小于'0'
     * （3）整数不仅1位，会大于10
     * （4）先做减法，避免int溢出
     * （5）char类型，不能使用switch
     *
     * @param s
     * @return
     */
    public int calculate(String s) {
        char sign = '+'; // 保存上一个符号，初始为'+'
        Deque<Integer> stack = new LinkedList<>();
        int num = 0, ans = 0;
        for (int i = 0; i < s.length(); i++) {
            char cur = s.charAt(i);
            if (cur >= '0') {
                num = num * 10 - '0' + cur; // 记录当前数字，先减，防溢出
            }
            if ((cur < '0' && cur != ' ') || 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 = cur; // sign更新为当前符号
                num = 0;    // 当前数字清零
            }
        }
        while (!stack.isEmpty()) {  // 将栈内剩余数字累加，即为结果
            ans += stack.pop();
        }
        return ans;
    }

本地测试

        /**
         * 227. 基本计算器 II
         */
        lay.showTitle(227);
        Solution227 sol227 = new Solution227();
        String s227 = " 3+5 / 2 ";
        System.out.println(s227 + " = " + sol227.calculate(s227));