题目
你现在是一场采用特殊赛制棒球比赛的记录员。这场比赛由若干回合组成,过去几回合的得分可能会影响以后几回合的得分。
比赛开始时,记录是空白的。你会得到一个记录操作的字符串列表 ops
,其中 ops[i]
是你需要记录的第 i
项操作,ops
遵循下述规则:
- 整数
x
- 表示本回合新获得分数x
"+"
- 表示本回合新获得的得分是前两次得分的总和。题目数据保证记录此操作时前面总是存在两个有效的分数。"D"
- 表示本回合新获得的得分是前一次得分的两倍。题目数据保证记录此操作时前面总是存在一个有效的分数。"C"
- 表示前一次得分无效,将其从记录中移除。题目数据保证记录此操作时前面总是存在一个有效的分数。
请你返回记录中所有得分的总和。
思路与算法
使用变长数组对栈进行模拟。
如果操作是 +,那么访问数组的后两个得分,将两个得分之和加到总得分,并且将两个得分之和入栈。
如果操作是 D,那么访问数组的最后一个得分,将得分乘以 2 加到总得分,并且将得分乘以 2 入栈。
如果操作是 C,那么访问数组的最后一个得分,将总得分减去该得分,并且将该得分出栈。
如果操作是整数,那么将该整数加到总得分,并且将该整数入栈。
代码
#include<vector>
#include<string>
#include<stdexcept>//用于抛出异常
using namespace std;
class Solution
{
public:
int calPoints(vector<string>& ops)
{
int ret=0;
vector<int>points;
for(const auto& op:ops)
{
int n=points.size();
if(n==0&&(op[0]=='+'||op[0]=='D'))
{
//如果points为空,但是尝试+或者D操作,则抛出异常
throw runtime_error("Invalid operation:not enough previous scores");
}
switch(op[0])
{
case '+':
ret+=points[n-1]+points[n-2];
points.push_back(points[n-1]+points[n-2]);
break;
case 'D':
ret+=2*points[n-1];
points.push_back(points[n-1]*2);
break;
case 'C':
if(n>0)
{
ret-=points[n-1];
points.pop_back();
}
break;
default:
ret+=stoi(op);
points.push_back(stoi(op));
break;
}
}
}
};
分析:
添加了一个检查来确保在尝试执行 '+' 或 'D' 操作之前,points
数组中有足够的元素。这是一个额外的安全措施,以防止在输入序列无效时程序崩溃或产生错误的结果。
另外,使用了 const auto& op : ops
来遍历 ops
向量可以避免不必要的字符串复制,并提高效率。在这个特定的例子中,由于 op
只是在 switch
语句中被读取,所以使用引用是安全的。
最后,请注意,虽然 ret -= points[n - 1];
在逻辑上是正确的(因为在 'C' 操作时需要从总分中减去最近添加的分数),但在某些情况下,如果只关心最终的总分而不关心 ret
在每个步骤中的值,那么也可以省略这行代码,因为 points.pop_back();
已经确保了 points
数组中的分数被正确地移除了,而 ret
的值将在循环结束时自动反映剩余的分数之和。然而,保留这行代码可以使得 ret
在每一步都保持最新状态,这在某些调试或日志记录场景中可能是有用的。