LeetCode 495. 提莫攻击

495. 提莫攻击

在《英雄联盟》的世界中，有一个叫 “提莫” 的英雄。他的攻击可以让敌方英雄艾希（编者注：寒冰射手）进入中毒状态。

当提莫攻击艾希，艾希的中毒状态正好持续 duration 秒。

正式地讲，提莫在 t 发起攻击意味着艾希在时间区间 [t, t + duration - 1]（含 t 和 t + duration - 1）处于中毒状态。如果提莫在中毒影响结束 前 再次攻击，中毒状态计时器将会 重置 ，在新的攻击之后，中毒影响将会在 duration 秒后结束。

给你一个 非递减 的整数数组 timeSeries ，其中 timeSeries[i] 表示提莫在 timeSeries[i] 秒时对艾希发起攻击，以及一个表示中毒持续时间的整数 duration 。

返回艾希处于中毒状态的 总 秒数。

示例 1：

输入：timeSeries = [1,4], duration = 2
输出：4
解释：提莫攻击对艾希的影响如下：
- 第 1 秒，提莫攻击艾希并使其立即中毒。中毒状态会维持 2 秒，即第 1 秒和第 2 秒。
- 第 4 秒，提莫再次攻击艾希，艾希中毒状态又持续 2 秒，即第 4 秒和第 5 秒。
艾希在第 1、2、4、5 秒处于中毒状态，所以总中毒秒数是 4 。

示例 2：

输入：timeSeries = [1,2], duration = 2
输出：3
解释：提莫攻击对艾希的影响如下：
- 第 1 秒，提莫攻击艾希并使其立即中毒。中毒状态会维持 2 秒，即第 1 秒和第 2 秒。
- 第 2 秒，提莫再次攻击艾希，并重置中毒计时器，艾希中毒状态需要持续 2 秒，即第 2 秒和第 3 秒。
艾希在第 1、2、3 秒处于中毒状态，所以总中毒秒数是 3 。

提示：

• 1 <= timeSeries.length <= 10^4
• 0 <= timeSeries[i], duration <= 10^7
• timeSeries 按 非递减 顺序排列

解法1：差分

Java版：

class Solution {
    public int findPoisonedDuration(int[] timeSeries, int duration) {
        int n = timeSeries.length;
        int maxTime = timeSeries[n - 1];
        int m = maxTime + duration;
        int[] diff = new int[m + 1];
        for (int t : timeSeries) {
            diff[t]++;
            diff[t + duration]--;
        }
        int ans = diff[0] > 0 ? 1 : 0;
        for (int i = 1; i < m; i++) {
            diff[i] += diff[i - 1];
            if (diff[i] > 0) {
                ans++;
            }
        }
        return ans;
    }
}

Python3版：

class Solution:
    def findPoisonedDuration(self, timeSeries: List[int], duration: int) -> int:
        maxTime = timeSeries[-1]
        m = maxTime + duration - 1
        diff = [0] * (m + 2)
        for t in timeSeries:
            diff[t] += 1
            diff[t + duration] -= 1
        
        ans = 1 if diff[0] > 0 else 0 
        for i in range(1, m + 2):
            diff[i] += diff[i - 1]
            if diff[i] > 0:
                ans += 1
        return ans

复杂度分析

• 时间复杂度：O(n + m)，其中 n 是数组 timeSeries 的长度, m 时差分数组 diff 的长度。
• 空间复杂度：O(m)。

解法2：模拟

我们只需要对数组进行一次扫描就可以计算出总的中毒持续时间。我们记录艾希恢复为未中毒的起始时间 expired，设艾希遭遇第 i 次的攻击的时间为 timeSeries[i]。当艾希遭遇第 i 攻击时：

• 如果当前他正处于未中毒状态，则此时他的中毒持续时间应增加 duration​，同时更新本次中毒结束时间 expired​ 等于 timeSeries[i]+duration​；
• 如果当前他正处于中毒状态，由于中毒状态不可叠加，我们知道上次中毒后结束时间为 expired​​，本次中毒后结束时间为 timeSeries[i]+duration​​，因此本次中毒增加的持续中毒时间为 timeSeries[i]+duration−expired​​；

我们将每次中毒后增加的持续中毒时间相加即为总的持续中毒时间。

Java版：

class Solution {
    public int findPoisonedDuration(int[] timeSeries, int duration) {
        int ans = 0;
        int expired = 0;
        for (int t : timeSeries) {
            if (t >= expired) {
                ans += duration;
            } else {
                ans += t + duration - expired;
            }
            expired = t + duration;
        }
        return ans;
    }
}

Python3版：

class Solution:
    def findPoisonedDuration(self, timeSeries: List[int], duration: int) -> int:
        ans = 0
        expired = 0
        for t in timeSeries:
            if t >= expired:
                ans += duration
            else:
                ans += t + duration - expired 
            expired = t + duration
        return ans

复杂度分析

• 时间复杂度：O(n)，其中 n 是数组 timeSeries 的长度。我们只需要遍历一遍数组即可，因此总时间复杂度为 O(n)。
• 空间复杂度：O(1)。只需要记录未中毒的起始时间即可，因此时间复杂度为 O(1)。