题目
在《英雄联盟》的世界中,有一个叫 “提莫” 的英雄,他的攻击可以让敌方英雄艾希(编者注:寒冰射手)进入中毒状态。现在,给出提莫对艾希的攻击时间序列和提莫攻击的中毒持续时间,你需要输出艾希的中毒状态总时长。
你可以认为提莫在给定的时间点进行攻击,并立即使艾希处于中毒状态。
示例1:
输入: [1,4], 2
输出: 4
原因: 第 1 秒初,提莫开始对艾希进行攻击并使其立即中毒。中毒状态会维持 2 秒钟,直到第 2 秒末结束。
第 4 秒初,提莫再次攻击艾希,使得艾希获得另外 2 秒中毒时间。
所以最终输出 4 秒。
示例2:
输入: [1,2], 2
输出: 3
原因: 第 1 秒初,提莫开始对艾希进行攻击并使其立即中毒。中毒状态会维持 2 秒钟,直到第 2 秒末结束。
但是第 2 秒初,提莫再次攻击了已经处于中毒状态的艾希。
由于中毒状态不可叠加,提莫在第 2 秒初的这次攻击会在第 3 秒末结束。
所以最终输出 3 。
解题思路
维护当前中毒状态的最右边界,如果在中毒状态下又被攻击,则延长中毒时间,否则重新计算中毒状态。
思路和划分字母区间的第二个思路很像。
当然这道题也可以转换为区间合并的题目:中毒状态可以看作是一个这样的区间:[被攻击的时间, 被攻击的时间 + 中毒时间]
,求出所有区间后,再对区间做合并即可。
代码
class Solution:
def findPoisonedDuration(self, timeSeries: List[int], duration: int) -> int:
poision_time = 0
prev_time = 0
right = 0
for time in timeSeries:
if time > right:
poision_time += (right - prev_time)
prev_time = time
right = time + duration
return poision_time + (right - prev_time)