https://leetcode-cn.com/problems/basic-calculator/
题目
实现一个基本的计算器来计算一个简单的字符串表达式的值。
字符串表达式可以包含左括号 ( ,右括号 ),加号 + ,减号 -,非负整数和空格 。
示例
输入
" 2-1 + 2 "
输出
3
输入
"(1+(4+5+2)-3)+(6+8)"
输出
23
一开始想过每次碰到括号就把括号的算出来再放到栈里面,但是写了一遍之后发现有各种问题。。以下为正确方法
思路
1.因为数字可能不止一位,所以while(isdigit(s[i])) ,把数字整个的大小算出来,存到num里。
2.定义一个标记flag,用来判断正负,碰到符号 (’+’ || ‘-’) 时,就把先前的flag乘上存好的num算出来加到答案里,新的flag据当前符号正负而定。
3. 前面还一直没有涉及到栈,当碰到括号时就需要栈了。我们可以 “拆括号” 。举个例子,随便写一个式子:1 - (5 +3 - 2) ,可以等效为1 - 5 - 3 + 2。由此很容易看出,括号前是正号,那么拆括号时数字的正负不会变,而若是负号,则会变为相反,因此我们只要在每次碰到 ‘(’ 时push进入当前的flag,当在括号中碰到前面的第二项,即碰到符号时,也只要在计算的时候等效的拆一下括号再加入答案即可,而多重括号的问题也因为栈的特性而被解决。
代码
class Solution {
public:
int calculate(string s) {
stack <int> f;
int flag = 1;
int num = 0;
int ans = 0;
f.push(1);
for (int i = 0; i < s.length(); i++) {
if (isdigit(s[i])) {
num = num *10 + (s[i] - 48);
}
if (s[i] == '+' || s[i] == '-') {
ans += num * flag;
flag = f.top();
flag *= s[i] == '+' ? 1 : -1;
num = 0;
}
if (s[i] == '(') f.push(flag);
if (s[i] == ')') f.pop();
}
return ans + flag * num;
}
};