实现一个基本的计算器来计算一个简单的字符串表达式 s
的值。
示例 1:
输入:s = "1 + 1"
输出:2
示例 2:
输入:s = " 2-1 + 2 "
输出:3
示例 3:
输入:s = "(1+(4+5+2)-3)+(6+8)"
输出:23
提示:
1 <= s.length <= 3 * 105
s
由数字、’+’、’-’、’(’、’)’、和 ’ ’ 组成s
表示一个有效的表达式
解答
由于这道题没有乘除,所以所有运算的优先级一致,即可以从左至右遍历运算。比较麻烦的一点在于括号的存在可能会使括号内的若干操作符加减翻转,如-(1+2-3)
展开后是-1-2+3
。可以看到产生这种反转的符号都位于括号左右。可以使用栈维护当前位置最靠近的左括号前的操作符(遇到左括号就将当前操作符入栈),随后,在遇到右括号前所有的操作符实际含义应该是栈顶操作符的正负乘以遍历到的操作符的正负。右括号遍历后,应该出栈。
class Solution {
public:
int calculate(string s) {
int op = 1;
int result = 0;
int len = s.size();
int i = 0;
vector<int> last_ops{op};
while(i < len){
if(s[i] == ' '){
i++;
}
else if(s[i] == '+'){
op = last_ops.back();
i++;
}
else if(s[i] == '-'){
op = -last_ops.back();
i++;
}
else if(s[i] == '('){
last_ops.push_back(op);
i++;
}
else if(s[i] == ')'){
last_ops.pop_back();
i++;
}
else{
int temp = i;
while(i < len && (s[i] <= '9' && s[i] >= '0'))
i++;
result += op * stoi(s.substr(temp, i - temp));
}
}
return result;
}
};