现基本计算器以计算简单表达式字符串。
表达式字符串可以包含左括号(和右括号)、加号+或减号、非负整数和空格。
表达式字符串只包含非负整数、+、-、*、/运算符、左括号和空格。整数除法应该截断为零。
您可以假定给定的表达式总是有效的。所有中间结果将在范围内[-2147483648,2147483647]
"1 + 1" = 2
" 6-4 / 2 " = 4
"2*(5+5*2)/3+(6/2+8)" = 21
"(2+6* 3+5- (3*14/7+2)*5)+3"=-12
思路:
其实把所有的计算都看成加法,无非是加正数,加负数的问题。
遇到乘除法我们就原地计算成一个新的被加数。
遇到括弧就原地递归得出一个新的被加数。
class Solution {
public int calculate(String s) {
//去空格
s = s.replaceAll(" ", "");
if (s.length() == 0) return 0;
Stack<Integer> stack = new Stack<>();//数据栈
char sign = '+'; //符号
int res = 0, pre = 0, i = 0;
while (i < s.length()) {
char ch = s.charAt(i);
//处理两位数的问题
if (Character.isDigit(ch)) {
pre = pre*10+(ch-'0');
}
//碰到左括号 就把括号里面当成一个 新的被加数
if (ch == '(') {
int j = findClosing(s.substring(i));
pre = calculate(s.substring(i+1, i+j));
i += j;
}
if (i == s.length()-1 || !Character.isDigit(ch)) {
//将所有的结果压栈 最后统一加起来
switch (sign) {
case '+':
stack.push(pre); break;
case '-':
stack.push(-pre); break;
case '*':
stack.push(stack.pop()*pre); break;
case '/':
stack.push(stack.pop()/pre); break;
}
pre = 0;
//记录当前的符号
sign = ch;
}
i++;
}
// 本质上说全都是加法
while (!stack.isEmpty()) res += stack.pop();
return res;
}
//删除所有的括号对,并返回右括号的位置
private int findClosing(String s) {
int level = 0, i = 0;
for (i = 0; i < s.length(); i++) {
if (s.charAt(i) == '(') level++;
else if (s.charAt(i) == ')') {
level--;
if (level == 0) break;
} else continue;
}
return i;
}
}