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