题解 | #四则运算#
四则运算
https://www.nowcoder.com/practice/9999764a61484d819056f807d2a91f1e
import java.io.BufferedReader; import java.io.IOException; import java.io.InputStreamReader; import java.util.Stack; public class Main { public static void main(String[] args) { BufferedReader r = new BufferedReader(new InputStreamReader(System.in)); String a; try { a = r.readLine(); } catch (IOException e) { throw new RuntimeException(e); } char[] chs = a.toCharArray(); int i = 0, j, l = chs.length; int c1; float n1, n2, t = 0f; boolean minus; while (i < l) {//将所有的中括号、大括号替换为小括号 if (chs[i] == '[' || chs[i] == '{') chs[i] = '('; if (chs[i] == ']' || chs[i] == '}') chs[i] = ')'; i++; } Stack<Float> nums = new Stack<>();//初始化栈,存放数字 Stack<Integer> calcs = new Stack<>();//初始化栈,存放符号 i = 0; while (i < l) { if (chs[i] == '(') {//ASCII码40 calcs.push(40); i++; continue; } if (chs[i] == ')') {//ASCII码41 while ((c1 = calcs.peek()) != '(') {//遇到反括回,两栈分别弹出运算符和数字进行计算,直到符号栈顶的元素是(;判断栈顶元素,因此用peek。 calcs.pop(); n2 = nums.pop();//最上面是n2 n1 = nums.pop();//其次是n1 t = cal(n1, n2, c1);//运算实质是n1?n2 nums.push(t); } calcs.pop();//计算完毕,将小括号弹出 i++; continue; } if (chs[i] == '+' || chs[i] == '-' || chs[i] == '*' || chs[i] == '/') { minus = i > 0 && chs[i - 1] == '(' && chs[i] == '-'; minus = minus || i == 0 && chs[i] == '-'; if (minus) nums.push(0f); if (calcs.isEmpty()) {//存储括号和运算符的栈是空 calcs.push(0 + chs[i]); i++; continue; } else { c1 = calcs.peek(); if (isHigh(c1, chs[i])) {//栈顶是+-,当前是*/,继续压入栈,栈顶是(,先压入 calcs.push(0 + chs[i]); } else {//栈顶和当前都是+-,开始计算;栈顶是*/,先弹出计算 c1 = calcs.pop(); n2 = nums.pop(); n1 = nums.pop(); t = cal(n1, n2, c1); nums.push(t);//将计算好的数字存入数字栈 i--;//继续判断,符号栈是否是空,直到是空,到if为true的判断,将当前符号压入栈 } i++; continue; } } if ((chs[i] - '0' | '9' - chs[i]) > 0) {//遇到字符是数字,计算完整数字的大小 j = i + 1; n1 = chs[i] - '0'; while (j < l) { if ((chs[j] - '0' | '9' - chs[j]) < 0) break; n1 *= 10; n1 += chs[j] - '0'; j++; } nums.push(n1);//将数字压入数字栈 i = j; continue; } i++; } while (!calcs.isEmpty()) {//符号栈中还不是空的,逆向计算得到最后结果 c1 = calcs.pop(); n2 = nums.pop(); n1 = nums.pop(); t = cal(n1, n2, c1); nums.push(t); } t = nums.pop(); System.out.print(Math.round(t));//结果四舍五入 } //计算方法,43——+,45——-,42——*,47——/,a和b分别为前面的数和后面的数 private static float cal(float a, float b, int ope) { switch (ope) { case 43: return a + b; case 45: return a - b; case 42: return a * b; case 47: return a / b; default: return 0; } } //是否高优先级,符号栈顶的元素是+-,当前字符也是+-,可以先计算栈顶的运算符号; //符号栈顶的元素是*/,乘除的优先级最高,不管当前运算符是什么,都可以先计算栈顶的运算符号,从数字栈中取值; //符号栈顶的元素是+-,当前字符是*/,优先级高于+-,要将当前的运算符压入符号栈; //符号栈顶的元素是(,要将当前的运算符压入符号栈 private static boolean isHigh(int c1, int c2) { boolean res = false; if (c1 == 43 || c1 == 45) { if (c2 == 42 || c2 == 47) res = true; } else if (c1 == 40) res = true; return res; } }