实现一个基本的计算器来计算一个简单的字符串表达式的值。
字符串表达式可以包含左括号 ( ,右括号 ),加号 + ,减号 -,非负整数和空格 。
示例 1:
输入: “1 + 1”
输出: 2
示例 2:
输入: " 2-1 + 2 "
输出: 3
示例 3:
输入: “(1+(4+5+2)-3)+(6+8)”
输出: 23
class Solution {
public:
int calculate(string s) {
int sLen = s.length();
if(sLen == 0) return 0;
for(int i = 0; i < sLen; )
{
if(s[i] == ' ')
i++;
else if(s[i] == '(' || s[i] == '+' || s[i] == '-')
{
opeSta.push(s[i++]);
}
else if(s[i] == ')')
{
opeSta.pop();
i++;
if(!opeSta.empty() && opeSta.top() != '(') //针对3+(5+2)
{
innerOperate();
}
}
else
{
long numRead = 0;
while(i < sLen && s[i] >= '0' && s[i] <= '9')
{
numRead = numRead * 10 - '0' + s[i]; //取出完整的数字元素
i++;
}
dataSta.push(numRead);
if(!opeSta.empty() && opeSta.top() != '(')
{
innerOperate();
}
}
}
return dataSta.top();
}
private:
void innerOperate()
{
if(dataSta.empty() || opeSta.empty()) return;
int valueTwo = dataSta.top();
dataSta.pop();
int valueOne = dataSta.top();
dataSta.pop();
char operatorCh = opeSta.top();
opeSta.pop();
switch(operatorCh){
case '+':
dataSta.push(valueOne + valueTwo);
break;
case '-':
dataSta.push(valueOne - valueTwo);
break;
default:
break;
}
}
private:
stack<int> dataSta; //存储数字
stack<char> opeSta; //存储符号
};
使用两个栈,一个存储数字,一个存储符号,1、遇到左括号和加减号时,插入到符号栈中;2、遇到右括号,先弹出符号栈顶元素(此时栈顶元素一定是左括号),弹出之后判断栈顶元素是否是左括号,不是则进行计算,每次计算的结果压入到数字栈中;3、若是数字,取出完整的数字元素压入到数字栈中,判断符号栈顶元素是否为左括号,不是则进行计算。