Teemo Attacking Python Solution with Tests

lcpy gen -n 495   # by problem number
lcpy gen -s teemo_attacking   # by problem name

Input: timeSeries = [1,4], duration = 2
Output: 4
Explanation: Teemo's attacks on Ashe go as follows:
- At second 1, Teemo attacks, and Ashe is poisoned for seconds 1 and 2.
- At second 4, Teemo attacks, and Ashe is poisoned for seconds 4 and 5.
Ashe is poisoned for seconds 1, 2, 4, and 5, which is 4 seconds in total.

Input: timeSeries = [1,2], duration = 2
Output: 3
Explanation: Teemo's attacks on Ashe go as follows:
- At second 1, Teemo attacks, and Ashe is poisoned for seconds 1 and 2.
- At second 2 however, Teemo attacks again and resets the poison timer. Ashe is poisoned for seconds 2 and 3.
Ashe is poisoned for seconds 1, 2, and 3, which is 3 seconds in total.

class Solution:
    # Time: O(n)
    # Space: O(1)
    def find_poisoned_duration(self, time_series: list[int], duration: int) -> int:
        total = 0
        for i, t in enumerate(time_series):
            if i + 1 < len(time_series):
                total += min(duration, time_series[i + 1] - t)
            else:
                total += duration
        return total