【C++刷题】力扣-#495-提莫攻击

题目描述

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

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

给你一个非递减的整数数组 timeSeries ，其中 timeSeries $i$ 表示提莫在 timeSeries $i$ 秒时对艾希发起攻击，以及一个表示中毒持续时间的整数 duration 。返回艾希处于中毒状态的总秒数。

示例

示例 1

bash 复制代码

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

示例 2

bash 复制代码

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

题解

初始化中毒时间：设置变量 totalSeconds 为 0，用于记录总的中毒秒数。
遍历攻击序列：从第一个攻击时间开始遍历 timeSeries。
○ 对于每次攻击，计算中毒状态的结束时间，即当前攻击时间加上 duration - 1。
○ 如果下一次攻击时间在当前中毒状态的持续时间内，则重置当前中毒状态的结束时间为下一次攻击时间加上 duration - 1。
○ 将每次中毒状态的持续时间加到 totalSeconds。
返回结果：返回 totalSeconds。

代码实现

cpp 复制代码

int findPoisonedDuration(vector<int>& timeSeries, int duration) {
    int totalSeconds = 0;
    int n = timeSeries.size();
    for (int i = 0; i < n; ++i) {
        int start = timeSeries[i];
        int end = start + duration - 1;
        if (i == n - 1 || timeSeries[i + 1] > end) {
            // 如果是最后一次攻击或下一次攻击时间在当前中毒状态结束后
            totalSeconds += duration;
        } else {
            // 如果下一次攻击在当前中毒状态持续时间内
            totalSeconds += timeSeries[i + 1] - start;
        }
    }
    return totalSeconds;
}

复杂度分析

● 时间复杂度：O(n)，其中 n 是攻击序列 timeSeries 的长度。我们需要遍历一次攻击序列。

● 空间复杂度：O(1)，因为我们只使用了常数个额外变量。

这个算法的优势在于它直接遍历攻击序列并计算中毒状态的持续时间，不需要额外的数据结构。