计算器

面试题 16.26. 计算器

给定一个包含正整数、加(+)、减(-)、乘(*)、除(/)的算数表达式(括号除外)，计算其结果。

表达式仅包含非负整数，+， - ，*，/ 四种运算符和空格 。 整数除法仅保留整数部分。

示例 1:

输入: "3+2*2"输出: 7

示例 2:

输入: " 3/2 "输出: 1

示例 3:

输入: " 3+5 / 2 "输出: 5

private Map<Character, Integer> map = new HashMap<>();

{
    map.put('/', 2);
    map.put('*', 2);
    map.put('-', 1);
    map.put('+', 1);
}

public int calculate(String s) {
    Stack<Integer> numStack = new Stack<>();
    Stack<Character> operatorStack = new Stack<>();
    for (int i = 0; i < s.length();) {
        char c = s.charAt(i);
        if (!isNumber(c)) {//不是数字
            if (c != ' ') {//考虑空格
                boolean flag = false;
                //while考虑1-2+3
                while (!operatorStack.isEmpty() && isPrior(operatorStack.peek(), c)) {
                    calculate(numStack, operatorStack);
                    flag = true;
                }
                if (flag) {
                    operatorStack.push(s.charAt(i));
                } else {
                    operatorStack.push(c);
                }
            }
            i++;
        } else {
            StringBuilder sb = new StringBuilder();
            while (i < s.length() && isNumber(s.charAt(i))) {//多位数字
                sb.append(s.charAt(i));
                i++;
            }
            numStack.push(Integer.parseInt(sb.toString()));
        }
    }
    while (!operatorStack.isEmpty()) {
        calculate(numStack, operatorStack);
    }
    return numStack.pop();
}

private int calculate(int num1, int num2, char c) {
    if (c == '+') {
        return num1 + num2;
    }
    if (c == '-') {
        return num1 - num2;
    }
    if (c == '*') {
        return num1 * num2;
    }
    return num1 / num2;
}

public void calculate(Stack<Integer> numStack, Stack<Character> opStack) {
    char op = opStack.pop();
    int num1 = numStack.pop();
    int num2 = numStack.pop();
    numStack.push(calculate(num2, num1, op));
}

private boolean isNumber(char c) {
    return c >= '0' && c <= '9';
}

private boolean isPrior(char c1, char c2) {
    return map.get(c1) >= map.get(c2);
}
`