题解 | #四则运算#
四则运算
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;
}
}
阿里云工作强度 694人发布
