495. 提莫攻击
题目
在《英雄联盟》的世界中,有一个叫 “提莫” 的英雄。他的攻击可以让敌方英雄艾希(编者注:寒冰射手)进入中毒状态。
当提莫攻击艾希,艾希的中毒状态正好持续 duration
秒。
正式地讲,提莫在 t
发起攻击意味着艾希在时间区间 [t, t + duration - 1]
(含 t
和 t + duration - 1
)处于中毒状态。如果提莫在中毒影响结束 前 再次攻击,中毒状态计时器将会 重置 ,在新的攻击之后,中毒影响将会在 duration
秒后结束。
给你一个 非递减 的整数数组 timeSeries
,其中 timeSeries[i]
表示提莫在 timeSeries[i]
秒时对艾希发起攻击,以及一个表示中毒持续时间的整数 duration
。
返回艾希处于中毒状态的 总 秒数。
提示:
1 <= timeSeries.length <= 104
0 <= timeSeries[i], duration <= 107
timeSeries
按 非递减 顺序排列
示例
示例 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 。
题解
class Solution {
public:
int findPoisonedDuration(vector<int>& timeSeries, int duration) {
int ans = 0, last = -1;
for(const auto& s : timeSeries){
int e = s + duration - 1;
ans += last < s ? duration : e - last;
last = e;
}
return ans;
}
};
代码的逻辑是这样的:
ans
用于累计毒药的总持续时间。last
用于记录上一个时间段结束的时间。- 遍历
timeSeries
中的每个时间段s
:- 计算当前时间段结束的时间
e
,即开始时间s
加上毒药持续时间duration
减去1秒(因为结束时间是非包含的)。 - 如果当前时间段的开始时间
s
大于上一个时间段的结束时间last
,说明没有重叠,那么毒药的持续时间就是duration
秒。 - 如果有重叠(即
last >= s
),那么毒药的持续时间就是当前时间段的结束时间e
减去上一个时间段的结束时间last
。 - 累加到
ans
中。 - 更新
last
为当前时间段的结束时间e
。
- 计算当前时间段结束的时间