题目在这:https://leetcode-cn.com/problems/teemo-attacking/
思路分析:
这道题看着蛮有意思的就拿来做了做。
思路还是蛮简单的。
遍历所给数组timeSeries[i]。
- timeSeries[i+1] - timeSeries[i] 大于所给的中毒持续时间(duration),则答案直接加上duration。
- 如果两者相减不大于 ,则答案直接加上两者相减的值。
注意一个小点,就是无论如何,当数组所给的最后一个时间中毒后,将完整的经历中毒时间。
例如:
timeSeries = [1,5,6,8] duration = 3。
- 遍历,5-1>duration 所以答案res+=duration
- 6-5 < 3 所以答案 res+= 5-4
- 8-6 < 3 所以答案 res+= 5-4
此时遍历结束,但注意还有一个 8秒的时候中毒没有计算到,此时中毒后无论如何都将经历全部中毒时间,所以答案应该再 加上一个duration。
所以可以索性res初始化的时候就直接等于 duration。
完整代码
class Solution:
def findPoisonedDuration(self, timeSeries: List[int], duration: int) -> int:
res = duration
for i in range(len(timeSeries)-1):
if timeSeries[i] + duration <= timeSeries[i+1]:
res += duration
if timeSeries[i] + duration > timeSeries[i+1]:
res += timeSeries[i+1] - timeSeries[i]
return res