【LeetCode - 772】基本计算器 III

1、题目描述

2、解题思路

  这道题对于掌握了逆波兰计算器的人来说简直是送分题，唯一需要注意的地方就是计算的中间结果范围可能出现超出 int 类型的情况。因此，在进行逆波兰计算的时候，使用 long 来计算。

  逆波兰计算器的原理是使用逆波兰表达式来计算出表达式的值，我们人类能够熟练使用的是中缀表达式，比如2×(9+6/3-5)+4就是一个中缀表达式，但是看到上面的简单计算器就知道处理起来很麻烦。于是有一种逆波兰计算器，计算是在逆波兰表达式（也叫做后缀表达式）的基础上。
  逆波兰计算器的计算过程为：从左到右扫描后缀表达式，遇到数字就入栈，遇到操作符就从栈弹出两个数字，然后计算得到的值继续入栈，继续扫描表达式，直到扫描完毕得到结果。
  从逆波兰计算器的扫描过程可以看到，过程特别简单，代码写起来也比较容易。但现在的难点在于：如何把中缀表达式转成后缀表达式？

  这个过程已经有大神给出来了，我们记下来即可：

  1、初始化两个栈：运算符栈s1和存储中间结果的栈s2；

  2、从左到右扫描中缀表达式；

  3、遇到操作数时，压入到栈s2；

  4、遇到运算符时：

    1）如果s1为空或s1栈顶为左括号"("，则压入到s1；

    2）不满足1），则和s1栈顶运算符比较优先级，高于，则压入s1；

    3）不满足1）和2），弹出s1栈顶运算符并压入到s2，再次回到2）。

  5、遇到右括号")“时，依此弹出s1并压入s2，直到遇到左括号”)"为止，此时丢掉一对括号；

  6、重复2-5，直到扫描完毕；

  7、将s2栈弹出压入到s1，然后s1弹出全部，弹出的顺序即为后缀表达式。

3、解题代码

class Solution {
    public int calculate(String s) {
        String expression = s.replaceAll(" ", "");   // 去除空格
        List<String> infix = expressionToList(expression);  // List
        List<String> suffix = infixToSuffix(infix); // 中缀转后缀
        Stack<String> stack = new Stack<>();    // 存储中间结果
        // 逆波兰计算器
        for (int i = 0; i < suffix.size(); i++) {
            String tmp = suffix.get(i);
            if (isOper(tmp)) {
                String num2 = stack.pop();
                String num1 = stack.pop();
                String reuslt = cal(num1, tmp, num2);
                stack.push(reuslt);
            } else { // 数字直接入栈
                stack.push(tmp);
            }
        }
        return Integer.parseInt(stack.pop());
    }

    /**
     * num1 和 num2 进行 oper 计算
     *
     * @param num1
     * @param oper
     * @param num2
     * @return
     */
    public String cal(String num1, String oper, String num2) {
        Long result = 0l;
        Long a = Long.parseLong(num1);
        Long b = Long.parseLong(num2);
        switch (oper) {
            case "+":
                result = a + b;
                break;
            case "-":
                result = a - b;
                break;
            case "*":
                result = a * b;
                break;
            case "/":
                result = a / b;
                break;
        }
        return String.valueOf(result);
    }

    /**
     * 中缀转后缀
     *
     * @param infix
     * @return
     */
    public List<String> infixToSuffix(List<String> infix) {
        List<String> suffix = new ArrayList<>();
        Stack<String> stack1 = new Stack<>();   // 只用于保存操作符
        Stack<String> stack2 = new Stack<>();   // 用于保存中间结果
        for (int i = 0; i < infix.size(); i++) {
            String tmp = infix.get(i);
            if (isOper(tmp)) {   // 操作符要根据情况来入栈 1
                if (stack1.isEmpty() || "(".equals(tmp)) {
                    // 如果 stack1 是空的，或者 tmp 是左括号，直接入栈
                    stack1.push(tmp);
                } else { // stack1 不是空的，且 tmp 不是左括号
                    if (")".equals(tmp)) {
                        // tmp 是右括号，则把 stack1 遇到左括号之前，全部倒入 stack2
                        while (!"(".equals(stack1.peek())) {
                            stack2.push(stack1.pop());
                        }
                        stack1.pop();   // 丢掉左括号
                    } else {
                        // tmp 是 +-*/ 其中之一
                        while (!stack1.isEmpty() && priority(stack1.peek()) >= priority(tmp)) {
                            // 在 tmp 能够碾压 stack1 的栈顶操作符之前，把 stack1 的栈顶操作符倒入 stack 2 中
                            stack2.push(stack1.pop());
                        }
                        // 离开 while 时，要么 stack1 已经倒空了，要么就是现在 tmp 可以压住 stack.peek() 了
                        stack1.push(tmp);
                    }
                }
            } else { // 操作数直接入栈 2
                stack2.push(tmp);
            }
        }

        // stack1 剩余操作符全部倒入 stack2
        while (!stack1.isEmpty()) {
            stack2.push(stack1.pop());
        }

        // stack2 的逆序才是结果，所以要再倒一次
        while (!stack2.isEmpty()) {
            stack1.push(stack2.pop());
        }

        // 现在 stack1 的元素倒出来的顺序就是后缀表达式
        while (!stack1.isEmpty()) {
            suffix.add(stack1.pop());
        }

        return suffix;
    }

    /**
     * 字符串转为中缀 List
     *
     * @param expression
     * @return
     */
    public List<String> expressionToList(String expression) {
        List<String> list = new ArrayList<>();
        int len = expression.length();
        String keepNum = "";
        for (int i = 0; i < len; i++) {
            char c = expression.charAt(i);
            if (isNum(c)) {
                if (i != len - 1 && isNum(expression.charAt(i + 1))) {
                    // 如果下一个字符也是数字
                    keepNum += c;
                } else {
                    // 当前是最后一个字符，或者下一个开始不是数字
                    keepNum += c;
                    list.add(keepNum);
                    keepNum = "";
                }
            } else {
                list.add(c + "");
            }
        }
        return list;
    }

    /**
     * 判断字符 c 是不是数字
     *
     * @param c
     * @return
     */
    public boolean isNum(char c) {
        return c >= '0' && c <= '9';
    }

    /**
     * 判断字符串 str 是不是操作符
     *
     * @param str
     * @return
     */
    public boolean isOper(String str) {
        return "+".equals(str) || "-".equals(str) || "*".equals(str) || "/".equals(str) ||
                "(".equals(str) || ")".equals(str);
    }

    /**
     * 返回操作符的优先级，+- 为 0，* / 为 1
     *
     * @return
     */
    public int priority(String oper) {
        if ("+".equals(oper) || "-".equals(oper)) {
            return 0;
        } else if ("*".equals(oper) || "/".equals(oper)) {
            return 1;
        } else {
            return -1;
        }
    }
}