题目链接:495. 提莫攻击
若在上次攻击结束 duration 内再次攻击,则增加的中毒时长是两次攻击的间隔时间;否则,增加的中毒时长是 duration。
class Solution:
def findPoisonedDuration(self, timeSeries: List[int], duration: int) -> int:
ans, prev = 0, float('-inf')
for val in timeSeries:
if val - prev > duration:
ans += duration
else:
ans += val - prev
prev = val
return ans