java - 227. 基本计算器 II

1、题目

给你一个字符串表达式 s ，请你实现一个基本计算器来计算并返回它的值。

整数除法仅保留整数部分。

你可以假设给定的表达式总是有效的。所有中间结果将在 [-231, 231 - 1] 的范围内。

注意：不允许使用任何将字符串作为数学表达式计算的内置函数，比如 eval() 。

2、思路

        整体思路就是遍历字符串的每一位：遇到乘除法就标记，然后运算并压入栈中。为了方便运算，遇到减法将数字的负数压入栈中。最终循环结束统一将数字栈中的结果相加。

        这其中涉及到一个问题，就是数字有多位的问题，这里我们用temp进行判断。如果遇到数字后面也是数字，就累加temp，直到后面不是数字，将前面所有位压入栈中。具体遍历过程如下：

1、如果是乘除号，令flag=2，入符号栈。

2、如果是减号，令flag=1，入符号栈，加号默认flag=0

3、如果是数字，且flag=0，直接入栈；

4、如果是数字，且flag=1，该数的负数入栈，flag重置0；

5、如果是数字，且flag=2，和栈顶数字，栈顶符号运算，结果入栈，flag重置0

3、代码

class Solution {
    public int calculate(String s) {
        Stack<Integer> numStack = new Stack<>();
        Stack<Character> strStack = new Stack<>();
        int flag = 0; // 用来标记数字前的运算符号
        int result = 0; 
        int temp = 1; // 用来累加多位数字的位数
        int len = s.length();
        for (int i = 0; i < len; i++) {
            char tempChar = s.charAt(i);
            if (tempChar == '+') strStack.push(tempChar);
            else if (tempChar == '-') {
                strStack.push(tempChar);
                flag = 1;
            } else if (tempChar == '*') {
                flag = 2;
                strStack.push(tempChar);
            } else if (tempChar == '/') {
                flag = 2;
                strStack.push(tempChar);
            } else if (tempChar == ' ') {
            } else { //凡是数字都会走到这里

                if (flag == 0) { //凡是加号都会走到这里
                    if (i == len - 1) {// 防止最后一位数字越界
                        numStack.push(Integer.parseInt(s.substring(i - temp + 1, i + 1)));
                        temp = 1;
                        break;
                    }
                    char tempCharNext = s.charAt(i + 1);
                    if (Character.isDigit(tempCharNext)) {// 下一位是数字累加并继续遍历
                        temp++;
                        continue;
                    } else { // 下一位不再是数字就压入栈中
                        numStack.push(Integer.parseInt(s.substring(i - temp + 1, i + 1)));
                        temp = 1;
                    }

                } else if (flag == 1) {//凡是减号都会走到这里
                    if (i == len - 1) {// 防止最后一位数字越界
                        numStack.push(-Integer.parseInt(s.substring(i - temp + 1, i + 1)));
                        temp = 1;
                        break;
                    }
                    char tempCharNext = s.charAt(i + 1);
                    if (Character.isDigit(tempCharNext)) {// 下一位是数字累加并继续遍历
                        temp++;
                        continue;
                    } else {// 下一位不再是数字就压入栈中
                        numStack.push(-Integer.parseInt(s.substring(i - temp + 1, i + 1)));
                        temp = 1;
                    }
                } else {//凡是乘除号都会走到这里
                    if (strStack.peek() == '*') {
                        if (i == len - 1) {// 防止最后一位数字越界
                            int num = numStack.pop();
                            numStack.push(num * Integer.parseInt(s.substring(i - temp + 1, i + 1)));
                            strStack.pop();
                            temp = 1;
                            break;
                        }
                        char tempCharNext = s.charAt(i + 1);
                        if (Character.isDigit(tempCharNext)) {// 下一位是数字累加并继续遍历
                            temp++;
                            continue;
                        } else {// 下一位不再是数字就压入栈中
                            int num = numStack.pop();
                            numStack.push(num * Integer.parseInt(s.substring(i - temp + 1, i + 1)));
                            strStack.pop();
                            temp = 1;
                        }

                    } else {
                        if (i == len - 1) {// 防止最后一位数字越界
                            int num = numStack.pop();
                            numStack.push(num / Integer.parseInt(s.substring(i - temp + 1, i + 1)));
                            strStack.pop();
                            temp = 1;
                            break;
                        }
                        char tempCharNext = s.charAt(i + 1);
                        if (Character.isDigit(tempCharNext)) {// 下一位是数字累加并继续遍历
                            temp++;
                            continue;
                        } else {// 下一位不再是数字就压入栈中
                            int num = numStack.pop();
                            numStack.push(num / Integer.parseInt(s.substring(i - temp + 1, i + 1)));
                            strStack.pop();
                            temp = 1;
                        }
                    }
                }
                flag = 0;
            }
        }
        if (strStack.isEmpty()) {  // 没有符号就不用累加了
            return numStack.pop();
        }
        while (!numStack.isEmpty()) { //累加所有乘除后的结果
            result += numStack.pop();
        }
        return result;
    }

}