题目描述
你现在是一场采用特殊赛制棒球比赛的记录员。这场比赛由若干回合组成,过去几回合的得分可能会影响以后几回合的得分。
比赛开始时,记录是空白的。你会得到一个记录操作的字符串列表 ops,其中 ops[i] 是你需要记录的第 i 项操作,ops 遵循下述规则:
整数 x - 表示本回合新获得分数 x "+" - 表示本回合新获得的得分是前两次得分的总和。题目数据保证记录此操作时前面总是存在两个有效的分数。 "D" - 表示本回合新获得的得分是前一次得分的两倍。题目数据保证记录此操作时前面总是存在一个有效的分数。 "C" - 表示前一次得分无效,将其从记录中移除。题目数据保证记录此操作时前面总是存在一个有效的分数。 请你返回记录中所有得分的总和。
分析
输入:Array, 所有操作的 item,可能是分数或者操作 输出:Number, 最终的总分
解题思路
可以看到,我们会对上一回合和两回合的数据操作,所以用栈这个结构是很理想的。
我们遍历记录着操作的数组,对每一项分类讨论。
如果是 +:我们先 pop 前两项,再把 3 个数字按照顺序 push 回去,
对于 C:我们直接 pop 前一项
对于 C:我们 pop 前一项,算出一个 double 它的值,再把这两个数字按照顺序 push 回去。
代码
/**
* @param {string[]} ops
* @return {number}
*/
var calPoints = function (ops) {
const stack = []
for (let i = 0; i < ops.length; i++) {
const item = ops[i]
if (item === "+") {
const item1 = stack.pop()
const item2 = stack.pop()
stack.push(item2, item1, item1 + item2)
} else if (item === "D") {
const item1 = stack.pop()
stack.push(item1, item1 * 2)
} else if (item === "C") {
stack.pop()
} else {
stack.push(parseInt(item))
}
}
return stack.reduce((prev, cur) => prev + cur)
}
复杂度
时间:O(N),需要遍历所有的 operation
空间:O(N),需要存储所有的分数