题目描述
在你现在是一场采用特殊赛制棒球比赛的记录员。这场比赛由若干回合组成,过去几回合的得分可能会影响以后几回合的得分。
比赛开始时,记录是空白的。你会得到一个记录操作的字符串列表 ops,其中 ops[i] 是你需要记录的第 i 项操作,ops 遵循下述规则:
- 整数 x - 表示本回合新获得分数 x
- “+” - 表示本回合新获得的得分是前两次得分的总和。题目数据保证记录此操作时前面总是存在两个有效的分数。
- “D” - 表示本回合新获得的得分是前一次得分的两倍。题目数据保证记录此操作时前面总是存在一个有效的分数。
- “C” - 表示前一次得分无效,将其从记录中移除。题目数据保证记录此操作时前面总是存在一个有效的分数。
请你返回记录中所有得分的总和。
示例
示例 1
输入:ops = ["5","2","C","D","+"]
输出:30
解释:
"5" - 记录加 5 ,记录现在是 [5]
"2" - 记录加 2 ,记录现在是 [5, 2]
"C" - 使前一次得分的记录无效并将其移除,记录现在是 [5].
"D" - 记录加 2 * 5 = 10 ,记录现在是 [5, 10].
"+" - 记录加 5 + 10 = 15 ,记录现在是 [5, 10, 15].
所有得分的总和 5 + 10 + 15 = 30
示例 2
输入:ops = ["5","-2","4","C","D","9","+","+"]
输出:27
解释:
"5" - 记录加 5 ,记录现在是 [5]
"-2" - 记录加 -2 ,记录现在是 [5, -2]
"4" - 记录加 4 ,记录现在是 [5, -2, 4]
"C" - 使前一次得分的记录无效并将其移除,记录现在是 [5, -2]
"D" - 记录加 2 * -2 = -4 ,记录现在是 [5, -2, -4]
"9" - 记录加 9 ,记录现在是 [5, -2, -4, 9]
"+" - 记录加 -4 + 9 = 5 ,记录现在是 [5, -2, -4, 9, 5]
"+" - 记录加 9 + 5 = 14 ,记录现在是 [5, -2, -4, 9, 5, 14]
所有得分的总和 5 + -2 + -4 + 9 + 5 + 14 = 27
示例 3
输入:ops = ["1"]
输出:1
题解
为了解决这个问题,我们可以使用一个栈(stack)来存储每一回合的得分。遍历操作列表,对于每个操作,执行相应的操作:
- 如果是整数,将其压入栈中。
- 如果是 “+”,则将栈中最后两个元素的和压入栈中。
- 如果是 “D”,则将栈中最后一个元素的两倍压入栈中。
- 如果是 “C”,则弹出栈中最后一个元素。
最后,将栈中的所有元素相加,即可得到所有得分的总和。
代码实现
int calPoints(vector<string>& operations) {
stack<int> scores;
for (auto& op : operations) {
if (op == "+") {
// 本回合新获得的得分是前两次得分的总和
int second = scores.top();
scores.pop();
int first = scores.top();
scores.push(second);
scores.push(first + second);
} else if (op == "D") {
// 本回合新获得的得分是前一次得分的两倍
scores.push(scores.top() * 2);
} else if (op == "C") {
// 前一次得分无效,将其从记录中移除
scores.pop();
} else {
// 本回合新获得的分数 x
scores.push(stoi(op));
}
}
// 计算所有得分的总和
int total = 0;
while (!scores.empty()) {
total += scores.top();
scores.pop();
}
return total;
}
复杂度分析
● 时间复杂度: O(n),其中 n 是 ops 数组的长度。我们需要遍历一次数组。
● 空间复杂度: O(n),最坏情况下,我们可能需要将所有的操作都压入栈中。