Leetcode 224. Basic Calculator

最新推荐文章于 2023-12-22 17:08:09 发布

原创最新推荐文章于 2023-12-22 17:08:09 发布 · 157 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#stack #string

Leetcode 专栏收录该内容

278 篇文章

订阅专栏

本文介绍了一种使用栈从后向前遍历字符串来解析和计算数学表达式的算法。该方法适用于包含加减运算及括号的表达式，通过逆波兰表示法的思想实现，能够高效处理复杂的算术表达。

在这里插入图片描述
方法1: 从后往前遍历string，用一个stack。详细解释直接参考lc官方解答1。时间复杂n，空间复杂n，n为string的长度。

class Solution {
    public int calculate(String s) {
        if (s.charAt(0) == '-') s = "0" + s;
        int operand = 0;
        int n = 0;
        Stack<Object> stack = new Stack<>();
        
        for(int i = s.length()-1; i >= 0; i--){
            char c = s.charAt(i);
            if(Character.isDigit(c)){
                operand = (int)Math.pow(10, n) * (int)(c - '0') + operand;
                n++;
            }else if(c != ' '){
                if(n != 0){
                    stack.push(operand);
                    n = 0;
                    operand = 0;
                }
                if(c == '('){
                    int res = evaluateExpr(stack);
                    stack.pop();
                    stack.push(res);
                }else{
                    stack.push(c);
                }
            }
        }
        if(n != 0) stack.push(operand);
        return evaluateExpr(stack);
    }
    
    public int evaluateExpr(Stack<Object> stack){
        int res = 0;
        if(!stack.isEmpty()){
            res = (int) stack.pop();
        }
        
        while(!stack.isEmpty() && !((char)stack.peek() == ')')){
            char sign = (char) stack.pop();
            if(sign == '+'){
                res += (int) stack.pop();
            }else{
                res -= (int) stack.pop();
            }
        }
        return res;
    }
}