题解:
- 初始化一个变量total_time来累计总的中毒时间,还有一个变量
end_time
来记录上一次中毒结束的时间。 - 遍历数组
timeSeries
,对于每个元素(即每次攻击):- 如果当前攻击时间小于或等于上次中毒结束时间
end_time
,则只需计算中毒状态的增量,这是由当前攻击时间和上次中毒结束时间的差值决定的。 - 如果当前攻击时间大于上次中毒结束时间,说明艾希在两次攻击间恢复了正常状态,我们可以直接加上完整的中毒持续时间。
- 如果当前攻击时间小于或等于上次中毒结束时间
- 更新
end_time
为当前攻击时间加上中毒持续时间。 - 最后,返回总的中毒时间
total_time
。
class Solution:
def findPoisonedDuration(self, timeSeries: List[int], duration: int) -> int:
end_time = -1
total_time = 0
for t in timeSeries:
if t < end_time:
total_time += (t + duration) - end_time
else:
total_time += duration
end_time = t + duration
return total_time