LeetCode 772. Basic Calculator III

原题链接在这里：https://leetcode.com/problems/basic-calculator-iii/

题目：

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 .

The expression string contains only non-negative integers, +, -, *, / operators , open ( and closing parentheses ) and empty spaces . The integer division should truncate toward zero.

You may assume that the given expression is always valid. All intermediate results will be in the range of [-2147483648, 2147483647].

Some examples:

"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

Note: Do not use the eval built-in library function.

题解：

Now the calculator involves + - * /  and ( ).

Here could separate + - * / into 2 groups. First calculate * /, then + -. 

+, o1 = 1. -, o1 = -1. 

*, o2 = 1. /, o2 = -1.

res = num1 + o1 * num2. num2 already has result with * /.

When encountering number, get the current number, and calculate on o2 and get num2.

When encountering (, push all the current result into stack and get reset values.

When encountering ), first get current number and then update num2 with * and /.

Note: corner case, - at index 0 or after (, need to update o1 and continue.

Time Complexity: O(n). n = s.length.

Space: O(n).

AC Java:

class Solution {
    public int calculate(String s) {
        if(s == null || s.length() == 0){
            return 0;
        }

        s = s.replaceAll("\\s+", "");
        int n = s.length();
        int num1 = 0;
        int o1 = 1;
        int num2 = 1;
        int o2 = 1;
        Stack<Integer> stk = new Stack<>();

        for(int i = 0; i<n; i++){
            char c = s.charAt(i);
            if(Character.isDigit(c)){
                int cur = 0;
                while(i < n && Character.isDigit(s.charAt(i))){
                    cur = cur * 10 + (s.charAt(i) - '0');
                    i++;
                }

                i--;
                num2 = o2 == 1 ? num2 * cur : num2 / cur;
            }else if(c == '*' || c == '/'){
                o2 = c == '*' ? 1 : -1;
            }else if(c == '('){
                stk.push(num1);
                stk.push(o1);
                stk.push(num2);
                stk.push(o2);

                num1 = 0;
                o1 = 1;
                num2 = 1;
                o2 = 1;
            }else if(c == '+' || c == '-'){
                if(c == '-' && (i == 0 || s.charAt(i-1) == '(')){
                    o1 = -1;
                    continue;
                }

                num1 = num1 + o1 * num2;
                o1 = (c == '+' ? 1 : -1);

                num2 = 1;
                o2 = 1;
            }else if(c == ')'){
                int cur = num1 + o1 * num2;
                o2 = stk.pop();
                num2 = stk.pop();
                o1 = stk.pop();
                num1 = stk.pop();

                num2 = o2 == 1 ? num2 * cur : num2 / cur;
            }
        }

        return num1 + o1 * num2;
    }
}

类似Basic Calculator, Basic Calculator II.