棒球比赛

标签: 数组 模拟

难度: Easy

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

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

  1. 整数 x - 表示本回合新获得分数 x
  2. "+" - 表示本回合新获得的得分是前两次得分的总和。题目数据保证记录此操作时前面总是存在两个有效的分数。
  3. "D" - 表示本回合新获得的得分是前一次得分的两倍。题目数据保证记录此操作时前面总是存在一个有效的分数。
  4. "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

 

提示:

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

Submission

运行时间: 12 ms

内存: 16.2 MB

class Solution:
    def calPoints(self, operations: List[str]) -> int:
        score_sum = 0
        tmp_op = []
        for i in range(len(operations)):
            ele = operations[i]
            if ele == "C" :
                score_sum -= tmp_op[-1]
                tmp_op.pop()
            elif ele == "D":
                var = 2 * tmp_op[-1]
                score_sum += var
                tmp_op.append(var)
            elif ele == "+":
                var = tmp_op[-1] + tmp_op[-2]
                score_sum += var
                tmp_op.append(var)
            else:
                score_sum += int(ele)
                tmp_op.append(int(ele))
        return score_sum

Explain

该题解使用一个栈(列表)来模拟整个计分过程。遍历给定的操作列表,根据不同的操作符执行相应的计分操作。对于整数,直接将其加入到总分中,并将其压入栈中;对于 'C' 操作,从总分中减去栈顶元素,并将栈顶元素弹出;对于 'D' 操作,将栈顶元素的两倍加入到总分中,并将其压入栈中;对于 '+' 操作,将栈顶两个元素的和加入到总分中,并将其压入栈中。最后返回总分即可。

时间复杂度: O(n)

空间复杂度: O(n)

class Solution:
    def calPoints(self, operations: List[str]) -> int:
        score_sum = 0  # 总分
        tmp_op = []  # 栈,用于存储计分过程中的中间结果
        
        for i in range(len(operations)):
            ele = operations[i]
            
            if ele == "C" :
                # 对于 'C' 操作,从总分中减去栈顶元素,并将栈顶元素弹出
                score_sum -= tmp_op[-1]
                tmp_op.pop()
            elif ele == "D":
                # 对于 'D' 操作,将栈顶元素的两倍加入到总分中,并将其压入栈中
                var = 2 * tmp_op[-1]
                score_sum += var
                tmp_op.append(var)
            elif ele == "+":
                # 对于 '+' 操作,将栈顶两个元素的和加入到总分中,并将其压入栈中
                var = tmp_op[-1] + tmp_op[-2]
                score_sum += var
                tmp_op.append(var)
            else:
                # 对于整数,直接将其加入到总分中,并将其压入栈中
                score_sum += int(ele)
                tmp_op.append(int(ele))
        
        return score_sum

Explore

虽然题目保证在执行'+'操作时栈中至少存在两个元素,但在实际开发中,为了增强代码的鲁棒性和可维护性,添加对栈长度的检查是一个好习惯。这可以防止未来题目变更或数据错误导致的程序异常。因此,即使题目保证了数据的前提条件,添加检查仍然是推荐的做法。

与'+'操作类似,尽管题目保证在执行`D`操作时栈中至少有一个元素,但在更通用的实现中,检查栈是否为空可以避免因数据错误或问题变更导致的程序崩溃。添加栈是否为空的检查是一个好的编程习惯,可以提高程序的健壮性。

在实际的软件开发中,处理输入数据的有效性是非常重要的。对于非法的操作符,应该在函数的开始对输入参数进行验证,如果发现非法输入,可以抛出异常或返回错误信息。这样可以确保程序的健壮性并防止未预期的行为。