leetcode 224. Basic Calculator

题目描述

Implement a basic calculator to evaluate a simple expression string.

The expression string may contain open ( and closing parentheses ), the plus + or minus sign -, non-negative integers and empty spaces .

Example 1:

Input: "1 + 1"
Output: 2

Example 2:

Input: " 2-1 + 2 "
Output: 3

Example 3:

Input: "(1+(4+5+2)-3)+(6+8)"
Output: 23

Note:

• You may assume that the given expression is always valid.
• Do not use the eval built-in library function.

思路：

我们用一个栈来辅助计算，用个变量sign来表示当前的符号，我们遍历给定的字符串s，如果遇到了数字，由于可能是个多位数，所以我们要用while循环把之后的数字都读进来，然后用sign*num来更新结果res；如果遇到了加号，则sign赋为1，如果遇到了符号，则赋为-1；如果遇到了左括号，则把当前结果res和符号sign压入栈，res重置为0，sign重置为1；如果遇到了右括号，结果res乘以栈顶的符号，栈顶元素出栈，结果res加上栈顶的数字，栈顶元素出栈。

实现：

/*
    * 1. 没有括号时正常运算，将结果存入res中
    2. 遇到左括号时将res及括号之前的运算符压栈
    3. 遇到右括号时将当前值乘以左括号之前的符号（stack中取出），并加上左括号前的运算结果（stack中取出）：
    stack中存入的是遇到左括号时①括号之前计算结果res，②括号之前的运算符
    */
    public int calculate(String s) {

        Stack<Integer> stk = new Stack<>();
        // sign为1，可以考虑第一个数字没有符号的情况
        int sign = 1;//记录运算数字之前的符号：+为1 -为-1，由此来和后面的数做运算
        int res = 0;
        for (int i = 0; i < s.length(); i++) {
            char ch = s.charAt(i);
            if (Character.isDigit(ch)) {
                int curSum = s.charAt(i) - '0';
                //输出的是多位数
                while (i + 1 < s.length() && Character.isDigit(s.charAt(i + 1)))
                    curSum = curSum * 10 + s.charAt(++i) - '0';
                // res等于符号位乘以本次结果和之前的res之和
                res += sign * curSum;
            } else if (ch == '+')
                sign = 1;
            else if (ch == '-')
                sign = -1;
                // 遇到左括号时将res及括号之前的运算符压栈
            else if (ch == '(') {
                // 因res存入栈中，需要将res置为0，取出的时候从栈取res计算
                stk.push(res);
                stk.push(sign);// 注意
                res = 0;// 注意
                // sign置为1，已把sign放入栈中，重置
                sign = 1;
            } else if (ch == ')') {
                System.out.println(stk.peek());
                res*= stk.pop();//()前运算结果的符号
                System.out.println(stk.peek());
                res+=stk.pop();//()前的运算结果
            }
        }
        return res;
    }

参考：

https://www.cnblogs.com/grandyang/p/4570699.html