LeetCode224-----基本计算器(双栈)

题目表述

给你一个字符串表达式 s ,请你实现一个基本计算器来计算并返回它的值。

注意:不允许使用任何将字符串作为数学表达式计算的内置函数,比如 eval() 。

示例:
输入:s = "(1+(4+5+2)-3)+(6+8)"
输出:23

双栈解法

使用两个栈numsStackopsStack,分别用来存数字和操作运算符。
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);
    }
}
posted @ 2022-04-16 19:18  YoungerWb  阅读(117)  评论(0编辑  收藏  举报