本人算法萌新,为秋招找工作开始磨炼算法,算法题均用python实现,如果我有哪些地方做的有问题的,还请大家不吝赐教.
1.题干
你现在是一场采用特殊赛制棒球比赛的记录员。这场比赛由若干回合组成,过去几回合的得分可能会影响以后几回合的得分。
比赛开始时,记录是空白的。你会得到一个记录操作的字符串列表
ops
,其中ops[i]
是你需要记录的第i
项操作,ops
遵循下述规则:
- 整数
x
- 表示本回合新获得分数x
"+"
- 表示本回合新获得的得分是前两次得分的总和。题目数据保证记录此操作时前面总是存在两个有效的分数。"D"
- 表示本回合新获得的得分是前一次得分的两倍。题目数据保证记录此操作时前面总是存在一个有效的分数。"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"
操作,题目数据保证记录此操作时前面总是存在一个有效的分数
2.思考
看到C操作是弹出,想到了用栈来模拟操作,一道简单题目,按照题干给出的规则来做就ok.
3.代码
from typing import List
class Solution:
def calPoints(self, operations: List[str]) -> int:
stack = []
for o in operations:
if o == '+':
x = stack[-1]
y = stack[-2]
stack.append(x + y)
elif o == 'D':
stack.append(stack[-1] * 2)
elif o == 'C':
stack.pop()
else:
stack.append(int(o))
return sum(stack)
if __name__ == "__main__":
solution = Solution()
ops = ["1"]
print(solution.calPoints(ops))
4.总结
注意到题目中除数字外的操作是计算当前回合得分的,对之前的得分并没有影响,最后求和即可.