整体思路:按照“符号-数字”处理字符串,将+/-运算看作数字的符号不单独运算,因此不需要考虑符号的优先级问题,在遇到非数字符号或者到达字符串结尾时处理上一个“符号-数字”组合。将括号内容看作一个数字,递归处理,遇到“)”表示当前是计算的括号内容,因此直接将结果返回,一定注意共用一个索引,用来跳过括号内容。
class Solution {
public:
int helper(string s, int& i) {
int n = s.size(), num = 0;
stack<int> nums;
char sign = '+';
for (; i < n; i++) {
if (isdigit(s[i]))
num = 10 * num + (s[i] - '0');
else if (s[i] == '(') {
num = helper(s, ++i);
++i;
}
if (!isdigit(s[i]) || i == n - 1) {
int num1;
switch (sign) {
case '+': nums.push(num); break;
case '-': nums.push(-num); break;
case '*':
num1 = nums.top();
nums.pop();
nums.push(num1 * num);
break;
case '/':
num1 = nums.top();
nums.pop();
nums.push(num1 / num);
break;
}
num = 0;
if (s[i] == '+' || s[i] == '-' || s[i] == '*' || s[i] == '/')
sign = s[i];
}
if (s[i] == ')') break;
}
int ans = 0;
while (!nums.empty()) {
ans += nums.top();
nums.pop();
}
return ans;
}
int calculate(string s) {
if(s.empty()) return 0;
int i = 0;
return helper(s, i);
}
};
上面是在遇到+、-、*、/时计算结果,其实更直观的是在解析完数字时计算。解析数字有两种情况:纯数字or括号表达式。
void calculate_base(stack<int> &num, int n, char ch) {
if (ch == '+' || ch == '-') {
int fu = ch == '-' ? -1 : 1;
num.push(n * fu);
} else {
int a = num.top();
num.pop();
if (ch == '*') {
a = a * n;
} else if (ch == '/') {
a = a/n;
}
num.push(a);
}
}
int calculate(std::string s) {
int i = 0;
char ch = '+'; //默认运算符是+
stack<int> num;//+-直接认为是符号不参与运算放到栈里;只在*/运算避免了运算符优先级问题
while (i < s.size()) {
//纯数字
if (s[i] >= '0' && s[i] <= '9') {
int j = i;
while (j < s.size() && s[j] >= '0' && s[j] <= '9') {
++j;
}
int n = atoi(s.substr(i, j-i).c_str());
calculate_base(num, n, ch);
i = j;
}
//括号表达式
else if (s[i] == '(') {
int j = i;
while (j < s.size() && s[j] != ')') {
++j;
}
int n = calculate(s.substr(i+1, j-i-1).c_str());
calculate_base(num, n, ch);
i = j+1;
}
//运算符
else {
ch = s[i];
++i;
}
}
int ret = 0;
while (num.size() > 0) {
ret += num.top();
num.pop();
}
return ret;
}