[路飞]棒球比赛

249 阅读2分钟

题目描述

你现在是一场采用特殊赛制棒球比赛的记录员。这场比赛由若干回合组成,过去几回合的得分可能会影响以后几回合的得分。

比赛开始时,记录是空白的。你会得到一个记录操作的字符串列表 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),需要存储所有的分数