Teemo Attacking — LeetCode 495 Python Solution

The problem

Our hero Teemo is attacking an enemy Ashe with poison attacks! When Teemo attacks Ashe, Ashe gets poisoned for a exactly duration seconds.

Example

Input

timeSeries = [1,4], duration = 2

Output

4

Explanation

Teemo's attacks on Ashe go as follows:

Python solution

class Solution:
    def findPoisonedDuration(self, timeSeries: List[int], duration: int) -> int:
        ans = duration
        for a, b in pairwise(timeSeries):
            ans += min(duration, b - a)
        return ans

Complexity

Related problems