495. 提莫攻击

exiaohu 于 2021-11-10 发布

题目链接: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