LeetCode224-----基本计算器(双栈)
题目表述
给你一个字符串表达式 s ,请你实现一个基本计算器来计算并返回它的值。
注意:不允许使用任何将字符串作为数学表达式计算的内置函数,比如 eval() 。
示例:
输入:s = "(1+(4+5+2)-3)+(6+8)"
输出:23
双栈解法
使用两个栈numsStack和opsStack,分别用来存数字和操作运算符。
1、从前往后遍历,首先需要将字符串s中的运算符去掉。
2、匹配到(,直接加入到opsStack栈中。
3、匹配到),弹栈,直到遇到第一个左括号,将numStack中的数字和opsStack中的操作弹出进行运算。
4、匹配到数字,从当前为位置循环向后取,直到不再是数字。
5、匹配到运算符,将运算符加入到opsStack中,加入之前可以先将栈内的都计算出来。
由于第一个数可能是负数以及(后面也可能是-号,所以可以先往操作数栈里面补0。
class Solution {
public int calculate(String s) {
s = s.replaceAll(" ", "");
Deque<Integer> numStack = new ArrayDeque<>();
numStack.push(0);
Deque<Character> ops = new ArrayDeque<>();
for (int i = 0; i < s.length(); i++) {
char c = s.charAt(i);
if(c == '('){
ops.push(c);
}else if(c == ')'){
while(ops.peek() != '('){
cal(numStack, ops);
}
ops.pop();
}else{
if(Character.isDigit(c)){
int num = 0;
while( i < s.length() && Character.isDigit(s.charAt(i))){
num = num*10 + (s.charAt(i) - '0');
i++;
}
numStack.push(num);
i--;
}else{
if(i > 0 && s.charAt(i - 1) == '('){
numStack.push(0);
}
while(!ops.isEmpty() && ops.peek() != '(') cal(numStack, ops);
ops.push(c);
}
}
}
while(!ops.isEmpty()) cal(numStack, ops);
return numStack.pop();
}
public void cal(Deque<Integer> nums, Deque<Character> ops){
if(nums.size() < 2 || ops.isEmpty()) return;
int a = nums.pop(), b = nums.pop();
char op = ops.pop();
nums.push(op == '+' ? a + b : b - a);
}
}