LeetCode.682 棒球比赛

485 阅读2分钟

这是我参与11月更文挑战的第29天,活动详情查看:2021最后一次更文挑战

题目描述:

682. 棒球比赛 - 力扣(LeetCode) (leetcode-cn.com)

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

比赛开始时,记录是空白的。你会得到一个记录操作的字符串列表 ops,其中 ops[i] 是你需要记录的第 i项操作,ops 遵循下述规则:

整数 x - 表示本回合新获得分数 x "+"- 表示本回合新获得的得分是前两次得分的总和。题目数据保证记录此操作时前面总是存在两个有效的分数。 "D" - 表示本回合新获得的得分是前一次得分的两倍。题目数据保证记录此操作时前面总是存在一个有效的分数。 "C" - 表示前一次得分无效,将其从记录中移除。题目数据保证记录此操作时前面总是存在一个有效的分数。 请你返回记录中所有得分的总和。

示例一

输入: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

示例二

输入: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

示例三

输入: ops = ["1"]
输出: 1

提示:

  • 1 <= ops.length <= 1000
  • ops[i] 为 "C"、"D"、"+",或者一个表示整数的字符串。整数范围是 [-3 * 10^4, 3 * 10^4]
  • 对于 "+" 操作,题目数据保证记录此操作时前面总是存在两个有效的分数
  • 对于 "C" 和 "D" 操作,题目数据保证记录此操作时前面总是存在一个有效的分数

思路分析

根据题意,我们只需要根据操作字符对最后的数据,因此栈是一个比较合适的数据结构。

AC代码

class Solution {
    fun calPoints(ops: Array<String>): Int {
        val stack = Stack<Int>()
        for (op in ops) {
            if (op.toIntOrNull() == null) {
                if (!stack.isEmpty()) {
                    when (op) {
                        "C" -> {
                            stack.pop()
                        }
                        "D" -> {
                            stack.push(stack.peek() * 2)
                        }
                        "+" -> {
                            if (stack.size >= 2) {
                                stack.push(stack.peek() + stack[stack.size - 2])
                            }
                        }
                    }
                }

            } else {
                stack.push(op.toInt())
            }
        }
        var result = 0
        stack.forEach { 
            result+=it
        }
        return result
    }
}

总结

每日简单一题。

参考

棒球比赛 - 棒球比赛 - 力扣(LeetCode) (leetcode-cn.com)