[LeetCode][JavaScript]Basic Calculator

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 .

You may assume that the given expression is always valid.

Some examples:

"1 + 1" = 2
" 2-1 + 2 " = 3
"(1+(4+5+2)-3)+(6+8)" = 23

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

https://leetcode.com/problems/basic-calculator/

 

 

---

 

 

这这这不是我每天都为之纠结的后缀表达式吗。

题目比较简单只有加减法，而且没有算上负数。

我赌5毛，将来会有一道hard就是这题带上负数，小数，大小括号，乘除法之类，名字我都起好了Basic Calculator II。

2个栈，第一个是放运算结果（或者中间运算结果）的栈resultStack，第二个是放运算符的栈opStack。

举栗子：1+2

1 --> resultStack : [1], opStack : []

+ --> resultStack : [1], opStack : [+]

2 --> resultStack : [1,2], opStack : [+]

压缩二元运算操作compress_2operators() --> resultStack : [3], opStack : []

结束

 

举栗子2： 3-(2-1)

3 --> resultStack : [3], opStack : []

- --> resultStack : [3], opStack : [-]

( --> resultStack : [3], opStack : [-,(]

2 --> resultStack : [3,2], opStack : [-,(]

- --> resultStack : [3,2], opStack : [-,(,-]

1 --> resultStack : [3,2,1], opStack : [-,(,-]

) --> 压缩括号操作compress_bracket()，中间会调用compress_2operators() --> resultStack : [3,1], opStack : [-]

压缩二元运算操作compress_2operators() --> resultStack : [2], opStack : []

结束

 

如果要加上乘除法，需要修改一下正则，加上乘除法和加减法的优先级。

 

/**
 * @param {string} s
 * @return {number}
 */
var calculate = function(s) {
    var resultStack = [];
    var opStack = [];
    var temp = "";
    for(var i = 0; i < s.length; i++){
        var ch = s[i];
        if(/^(\+|\-)$/.test(ch)){ // + -
            compress_2operators();
            opStack.push(ch);
        } else if (ch === '('){
            opStack.push(ch);
        } else if (ch === ')'){
            compress_bracket();
        } else if (/^[0-9]$/.test(ch)){
            temp += ch;
        }

        if(s[i + 1] && /^(\+|\-|\(|\))$/.test(s[i + 1])){ // + - ( )
            if(temp !== ""){
                resultStack.push(parseInt(temp));
                temp = "";
            }
        }
    }
    if(temp !== ""){
        resultStack.push(parseInt(temp));
        temp = "";
    }
    compress_2operators();
    return resultStack.pop();

    function compress_bracket(){
        while(opStack[opStack.length - 1] !== '('){
            compress_2operators();
        }
        opStack.pop(); //(
    }
    function compress_2operators(){
        while(/^(\+|\-)$/.test(opStack[opStack.length - 1])){ // + -
            var op = opStack.pop();
            var right = resultStack.pop();
            var left = resultStack.pop();
            if(op === '+'){
                resultStack.push(left + right);
            }else{
                resultStack.push(left - right);
            }
        }
    }
};