原题地址239. 滑动窗口最大值
给定一个数组 nums,有一个大小为 k 的滑动窗口从数组的最左侧移动到数组的最右侧。你只可以看到在滑动窗口 k 内的数字。滑动窗口每次只向右移动一位。
返回滑动窗口最大值。
示例:
输入: nums = [1,3,-1,-3,5,3,6,7], 和 k = 3
输出: [3,3,5,5,6,7]
解释:
滑动窗口的位置 最大值
--------------- -----
[1 3 -1] -3 5 3 6 7 3
1 [3 -1 -3] 5 3 6 7 3
1 3 [-1 -3 5] 3 6 7 5
1 3 -1 [-3 5 3] 6 7 5
1 3 -1 -3 [5 3 6] 7 6
1 3 -1 -3 5 [3 6 7] 7
注意:
你可以假设 k 总是有效的,1 ≤ k ≤ 输入数组的大小,且输入数组不为空。
进阶:
你能在线性时间复杂度内解决此题吗?
算法
队列 deque 记录极大值的下标。逐个遍历,对当前点 nums[i] ,若不比队尾大则进入 deque ;否则,队尾出队直到队尾不小于 nums[i] , nums[i] 入队。
若队头的下标和 i 的距离超过 k ,即位于窗口之外,出队。
当 i 不小于 k-1 的时候,就需要输出队头元素。
代码
def fun(nums, k):
"""
:type nums: List[int]
:type k: int
:rtype: List[int]
"""
ans = []
deque = []
for i in range(len(nums)):
while len(deque) > 0 and nums[deque[-1]] < nums[i]:
deque.pop()
deque.append(i)
if i >= deque[0] + k:
deque.pop(0)
if i >= k - 1:
ans.append(nums[deque[0]])
return ans