解题思路:
阅读完题目就能明白这是最简单的栈操作处理,直接初始化一个栈,按照题目的要求进行操作即可,代码如下:
class Solution {
public:
int calPoints(vector<string>& ops) {
stack<int> s;
for(int i = 0; i < ops.size(); i ++) {
if(ops[i] == "+") {
int first = s.top();
s.pop();
int second = s.top();
s.push(first);
s.push(first + second);
} else if(ops[i] == "D") {
s.push(2 * s.top());
} else if(ops[i] == "C") {
s.pop();
} else {
s.push(stoi(ops[i]));
}
}
int ans = 0;
while(!s.empty()) {
ans += s.top();
s.pop();
}
return ans;
}
};
但是仔细观察你就能发现,其实这其中最多只涉及栈最后两个元素的操作,那不如定义成数组,模拟栈的操作,入栈的过程中统计栈中元素和,思路也是同样的,代码如下:
class Solution {
public:
int calPoints(vector<string>& ops) {
vector<int> s;
int ans = 0;
for(int i = 0; i < ops.size(); i ++) {
int n = s.size();
if(ops[i] == "+") {
ans += s[n - 1] + s[n - 2];
s.push_back(s[n - 1] + s[n - 2]);
} else if(ops[i] == "D") {
ans += 2 * s[n - 1];
s.push_back(2 * s[n - 1]);
} else if(ops[i] == "C") {
ans -= s[n - 1];
s.pop_back();
} else {
ans += stoi(ops[i]);
s.push_back(stoi(ops[i]));
}
}
return ans;
}
};