法一(栈)
/**
* 法一(栈)
* 思路:
* (1)一般需要符号栈、数据栈两个栈
* (2)这里只用了一个数据栈,符号栈用一个变量sign代替了,只存储上一个符号
* 注意
* (1)表达式中没有括号
* (2)加减乘除空格的ASCII码都小于'0'
* (3)整数不仅1位,会大于10
* (4)先做减法,避免int溢出
* (5)char类型,不能使用switch
*
* @param s
* @return
*/
public int calculate(String s) {
char sign = '+'; // 保存上一个符号,初始为'+'
Deque<Integer> stack = new LinkedList<>();
int num = 0, ans = 0;
for (int i = 0; i < s.length(); i++) {
char cur = s.charAt(i);
if (cur >= '0') {
num = num * 10 - '0' + cur; // 记录当前数字,先减,防溢出
}
if ((cur < '0' && cur != ' ') || i == s.length() - 1) {
switch (sign) { // 判断上一个符号是什么
case '+':
stack.push(num); // 当前符号前的数字直接压栈
break;
case '-':
stack.push(-num); // 当前符号前的数字取反压栈
break;
case '*':
stack.push(stack.pop() * num); // 数据栈栈顶数字出栈,与当前符号前的数字相乘,结果值压栈
break;
case '/':
stack.push(stack.pop() / num); // 数据栈栈顶数字出栈,除于当前符号前的数字,结果值压栈
break;
}
sign = cur; // sign更新为当前符号
num = 0; // 当前数字清零
}
}
while (!stack.isEmpty()) { // 将栈内剩余数字累加,即为结果
ans += stack.pop();
}
return ans;
}
本地测试
/**
* 227. 基本计算器 II
*/
lay.showTitle(227);
Solution227 sol227 = new Solution227();
String s227 = " 3+5 / 2 ";
System.out.println(s227 + " = " + sol227.calculate(s227));