一、题目
二、分析
本来计划采用栈的方式,遇到+, -
就入栈,遇到*, /
就把栈顶元素取出来计算。但是数据类型很复杂,不好存入栈中。
采用记录符号的方式。
3+2*4
---------入栈过程----------
3, 入栈
+, 记录该符号为preSign, 即前一个符号。
2, 计算得num=2,此时会判断preSign ,为+,继续将2入栈。
*, 记录到preSign 中。
4, 计算得num=4,此时也会判断preSign, 为*,将栈顶元素,即将2取出来与现在的 num(4) 相乘,再放入栈中。
---------栈的变化:--------
3
3 2
3 8 // 取 2 与 num(4) 相乘,并替换原来的2.
最后只需要将栈中所有元素求和。
三、代码
class Solution {
public:
int calculate(string s) {
vector<int> s1;
int n(s.length());
long num = 0;
char preSign = '+';
for(int i = 0; i < n; i ++){
// 求整数num
while(i < n && s[i] >= '0' && s[i] <= '9'){
num = 10 * num + s[i] - '0';
i++;
}
if (!isdigit(s[i]) && s[i] != ' ' || i == n - 1) {
switch(preSign){
case '+': s1.push_back(num);break;
case '-': s1.push_back(-num);break;
case '*': s1.back() *= num;break;
case '/': s1.back() /= num;break;
}
// 记录数字的前一个符号
preSign = s[i];
num = 0;
}
}
return accumulate(s1.begin(), s1.end(), 0);
}
};
执行用时:8 ms, 在所有 C++ 提交中击败了91.51%的用户
内存消耗:8.8 MB, 在所有 C++ 提交中击败了30.24%的用户
四、附录
accumulate定义在#include中,作用有两个,一个是累加求和,另一个是自定义类型数据的处理。
accumulate(s1.begin(), s1.end(), 0)
头两个形参指定要累加的元素范围,第三个形参则是累加的初值。
vector<int> a = {1, 4, 5, 6, 10};
cout << accumulate(a.begin(), a.end(), 2) << endl;
// 对于字符串也可以
vector<string> b = {"ss", "re", "iii"};
cout << accumulate(b.begin(), b.end()-1, string("22")) << endl;
-------- result ---------
28
22ssre
参考:https://blog.csdn.net/qq_40803710/article/details/80273811