class Solution(object):
def findPoisonedDuration(self, timeSeries, duration):
"""
:type timeSeries: List[int]
:type duration: int
:rtype: int
"""
ans = 0
n = len(timeSeries)
for i in range(n - 1):
ans += min(duration, timeSeries[i + 1] - timeSeries[i])
ans += duration
return ans