leetcode 224 基本计算器

实现一个基本的计算器来计算一个简单的字符串表达式的值。

字符串表达式可以包含左括号 ( ，右括号 )，加号 + ，减号 -，非负整数和空格 。

示例 1:

输入: “1 + 1”
输出: 2
示例 2:

输入: " 2-1 + 2 "
输出: 3
示例 3:

输入: “(1+(4+5+2)-3)+(6+8)”
输出: 23

采用双栈法，设立一个数据栈和一个操作符栈，在遍历字符串的过程中，如果一个字符为数字字符，先将字符串中的数据找到，此处需要回退操作，然后将数据Push入栈，判断是否满足运算的条件。如果为左括号或者加减符号则直接入栈，遇到右括号先pop掉符号栈的栈顶元素，因为有一个右括号肯定会有一个左括号与之对应，然后再进行计算。
建立一个函数compute(),该函数的作用是处理数据栈中的数据，即如果数据栈中的数据小于2，就return,如果数据栈中的数据大于2，并且符号栈的栈顶元素为+号和-号，就依次弹出两个数据num1和num2,进行加减运算，并且将运算结果再次push进入数据栈，如此进行循环，如果符号栈的栈顶为’('符号则直接进行返回，什么也不做。

class Solution {
public:
    int calculate(string s) {
       stack<int> number_stack;
       stack<char> operation_stack;
       int number;
       for(int i = 0; i < s.size(); i++){
           if(s[i] == ' '){     //消去空格
               continue;
           }
           //处理数字，生成整数，并压栈，判断进行计算
           if(s[i] >= '0' && s[i] <= '9'){
                number = 0;
                while(s[i] >= '0' && s[i] <= '9'){
                    number = number * 10 + (s[i] - '0');
                    i++; 
                }
                i--;
                number_stack.push(number);
                compute(number_stack,operation_stack);
           }
           else if(s[i] == ')'){
               operation_stack.pop();
               compute(number_stack,operation_stack);
           }
           //如果为（+-运算符号则直接入栈
           else{
               operation_stack.push(s[i]);
           }
       }
       return number_stack.top();
    }
private:
    void compute(stack<int> &number_stack,stack<char> &operation_stack){
        if (number_stack.size() < 2){
			return;
		}
        while(operation_stack.size() > 0 && (operation_stack.top() == '+' || operation_stack.top() == '-')){
		int num2 = number_stack.top();
		number_stack.pop();
		int num1 = number_stack.top();
		number_stack.pop();
		if (operation_stack.top() == '+'){
			number_stack.push(num1 + num2);
		}
		else if(operation_stack.top() == '-'){
			number_stack.push(num1 - num2);
		}
		operation_stack.pop();
        }
    }
};