·题目描述
给你一个整数数组 nums,有一个大小为 k 的滑动窗口从数组的最左侧移动到数组的最右侧。你只可以看到在滑动窗口内的 k 个数字。滑动窗口每次只向右移动一位。
返回 滑动窗口中的最大值 。
示例 1:
输入: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
示例 2:
输入:nums = [1], k = 1
输出:[1]
·解题思路
1.创建一个非增双端队列(使用collection),用来存放最大值
2.循环窗口的长度为k, 也就是两个指标,i,和j(i表示开端,就表示末端)。用来遍历数组
3.判断条件。判断条件有三个:
—滑动窗口,如果左端滑出的值为q队列中的最大值,那么需要在滑动后将队列左端的数值弹出
—如果队列存在,并且窗口的值大于队列末端值,将队列的末端数值弹出。再添加数值到队列中
—当窗口左端的指标从0 开始的时候,就可以向结果数组中添加q队列的初始值。
·代码
import collections
class Solution:
def maxSlidingWindow(self, nums, k: int):
n = len(nums)
q = collections.deque()
result = set()
for i, j in zip(range(1-k,n-k+1),range(0,n)):
if i > 0 and nums[i-1] == q[0]:
q.popleft()
while q and nums[j]> q[-1]:
q.pop()
q.append(nums[j])
if i >= 0:
result.add(q[0])
return result