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);
    }
}


全部评论

相关推荐

尊嘟假嘟点击就送:加v细说,问题很大
点赞 评论 收藏
分享
点赞 收藏 评论
分享
牛客网
牛客企业服务