HJ50 四则运算 | 题解

可以把加减乘除分成2组，分别是+-和*/

然后用o1和o2来储存当前符号，具体定义为

+：o1=1
-：o1=-1

*：o2=1
/：o2=-1

通过表达式res=num1+o1*num2来保存当前算式，其中num1为已经计算部分的值，而num2是正在计算部分的值，num1和num2中间由+-隔开，其中num2正在通过*/计算出结果

当遇到数字时，直接获取数字，并且计算o2，获得num2

当遇到+-时，代表前面的公式已经计算完毕，计算num1，并更新o1，o2，num2

当遇到*/时，需要对num2进行更新

当遇到左括号时，把现在所有结果入栈，并且重置参数

当遇到右括号时，获得当前数字来更新num2

import java.util.Scanner;
import java.util.Stack;

public class Main {
    public static void main(String[] args) {
        Scanner in = new Scanner(System.in);
        String str = in.nextLine();
        int len = str.length();
        int num1 = 0, num2 = 1;
        int o1 = 1, o2 = 1;
        Stack<Integer> stk = new Stack<>();

        for (int i = 0; i < len; i++) {
            char ch = str.charAt(i);
            if (Character.isDigit(ch)) {
                int cur = 0;
                while (i < len && Character.isDigit(str.charAt(i))) {
                    cur = cur * 10 + (str.charAt(i) - '0');
                    i++;
                }
                i--;
                num2 = o2 == 1 ? num2 * cur : num2 / cur;
            } else if (ch == '*' || ch == '/') {
                o2 = ch == '*' ? 1 : -1;
            } else if (ch == '(' || ch == '{' || ch == '[') {
                stk.push(num1);
                stk.push(o1);
                stk.push(num2);
                stk.push(o2);

                num1 = 0;
                o1 = 1;
                num2 = 1;
                o2 = 1;
            } else if (ch == '+' || ch == '-') {
                if (ch == '-' && (i == 0 || str.charAt(i - 1) == '(' || str.charAt(i - 1) == '[' || str.charAt(i - 1) == '{')) {
                    o1 = -1;
                    continue;
                }
                num1 = num1 + o1 * num2;
                o1 = (ch == '+' ? 1 : -1);
                num2 = 1;
                o2 = 1;
            } else {
                int cur = num1 + o1 * num2;
                o2 = stk.pop();
                num2 = stk.pop();
                o1 = stk.pop();
                num1 = stk.pop();

                num2 = o2 == 1 ? num2 * cur : num2 / cur;
            }
        }

        System.out.println(num1 + o1 * num2);
    }
}