在《英雄联盟》的世界中,有一个叫 “提莫” 的英雄,他的攻击可以让敌方英雄艾希(编者注:寒冰射手)进入中毒状态。现在,给出提莫对艾希的攻击时间序列和提莫攻击的中毒持续时间,你需要输出艾希的中毒状态总时长。
你可以认为提莫在给定的时间点进行攻击,并立即使艾希处于中毒状态。
示例1:
输入: [1,4], 2 输出: 4 原因: 在第 1 秒开始时,提莫开始对艾希进行攻击并使其立即中毒。中毒状态会维持 2 秒钟,直到第 2 秒钟结束。 在第 4 秒开始时,提莫再次攻击艾希,使得艾希获得另外 2 秒的中毒时间。 所以最终输出 4 秒。
示例2:
输入: [1,2], 2 输出: 3 原因: 在第 1 秒开始时,提莫开始对艾希进行攻击并使其立即中毒。中毒状态会维持 2 秒钟,直到第 2 秒钟结束。 但是在第 2 秒开始时,提莫再次攻击了已经处于中毒状态的艾希。 由于中毒状态不可叠加,提莫在第 2 秒开始时的这次攻击会在第 3 秒钟结束。 所以最终输出 3。
注意:
- 你可以假定时间序列数组的总长度不超过 10000。
- 你可以假定提莫攻击时间序列中的数字和提莫攻击的中毒持续时间都是非负整数,并且不超过 10,000,000。
第一种思路:
初始化一个res数组全为0,如果某一时刻i,提莫攻击了艾希,则把res[i]到res[i + duration -1]之间全部置为1,
最后返回sum(res)即可,
这种思路的问题在于题目数据规模有点大,过不了OJ
class Solution(object):
def findPoisonedDuration(self, timeSeries, duration):
"""
:type timeSeries: List[int]
:type duration: int
:rtype: int
"""
timeSeries.sort()
res = [0] * 200000100
for i in range(len(timeSeries)):
for j in range(duration):
res[timeSeries[i] + j] = 1
return sum(res)
第二种思路:
线性扫描,扫描到提莫攻击的每个时刻时,先判断一下下一次攻击时艾希还会不会处在中毒的状态,
如果艾希已经不再中毒,那么res 直接加上duration即可,
如果艾希还在中毒,那么res就加上两个时刻之差,代表中毒状态刷新,
最后记得加上最后一次射击带来的中毒持续时间(如果存在最后一次射击的话)。
class Solution(object):
def findPoisonedDuration(self, timeSeries, duration):
"""
:type timeSeries: List[int]
:type duration: int
:rtype: int
"""
# timeSeries.sort()
# res = [0] * 30000000
# for i in range(len(timeSeries)):
# for j in range(duration):
# res[timeSeries[i] + j] = 1
# return sum(res)
# timeSeries.sort()
res = 0
l = len(timeSeries)
for i in range(l - 1):
if timeSeries[i] + duration <= timeSeries[i+1]:
res += duration
else:
res += timeSeries[i+1] - timeSeries[i]
return res + duration if l else res