LeetCode495 提莫攻击

LeetCode495 提莫攻击

贪心维护最右

class Solution:
    def findPoisonedDuration(self, timeSeries: List[int], duration: int) -> int:

        time, n, r = 0, len(timeSeries), 0
        for i in range(n):

            if timeSeries[i] >= r: time, r = time + duration, timeSeries[i] + duration
            else: time, r = time + max(0, timeSeries[i] + duration - r), max(r, timeSeries[i] + duration)
        
        return time