Question

495. 提莫攻击

English Version

题目描述

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

当提莫攻击艾希，艾希的中毒状态正好持续 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 <= 104
• 0 <= timeSeries[i], duration <= 107
• timeSeries 按 非递减 顺序排列

解法

方法一：一次遍历

我们先考虑最后一次攻击，此次攻击一定可以使得艾希处于中毒状态，所以总中毒时间至少为 duration。

接下来，我们考虑前 $n-1$ 次攻击，每一次攻击的中毒持续时间为 $min(duration, timeSeries[i] - timeSeries[i-1])$，其中 $i$ 从 1 开始。我们将这些中毒持续时间累加起来，即为总中毒时间。

时间复杂度 $O(n)$，空间复杂度 $O(1)$。其中 $n$ 为数组 timeSeries 的长度。

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

class Solution {
  public int findPoisonedDuration(int[] timeSeries, int duration) {
    int n = timeSeries.length;
    int ans = duration;
    for (int i = 1; i < n; ++i) {
      ans += Math.min(duration, timeSeries[i] - timeSeries[i - 1]);
    }
    return ans;
  }
}

class Solution {
public:
  int findPoisonedDuration(vector<int>& timeSeries, int duration) {
    int ans = duration;
    int n = timeSeries.size();
    for (int i = 1; i < n; ++i) {
      ans += min(duration, timeSeries[i] - timeSeries[i - 1]);
    }
    return ans;
  }
};

func findPoisonedDuration(timeSeries []int, duration int) (ans int) {
  ans = duration
  for i, x := range timeSeries[1:] {
    ans += min(duration, x-timeSeries[i])
  }
  return
}

function findPoisonedDuration(timeSeries: number[], duration: number): number {
  const n = timeSeries.length;
  let ans = duration;
  for (let i = 1; i < n; ++i) {
    ans += Math.min(duration, timeSeries[i] - timeSeries[i - 1]);
  }
  return ans;
}

public class Solution {
  public int FindPoisonedDuration(int[] timeSeries, int duration) {
    int ans = duration;
    int n = timeSeries.Length;
    for (int i = 1; i < n; ++i) {
      ans += Math.Min(duration, timeSeries[i] - timeSeries[i - 1]);
    }
    return ans;
  }
}