引入
本题🔗看起来十分的简单:
给定一个数组 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,要求出范围k内的最大值,如果用暴力解法,在范围k的数组内进行依次比较,复杂度是O(kN),最坏情况k==N的时候,复杂度就是O(N^2)。
所以,此次要介绍一种之前没有接触到的数据结构,双端队列。双端队列其实很简单,比较于普通队列的先进先出,双端队列就是双端都可以出入。
那么怎么利用双端队列来解决本题呢?
本题解法
主要思路是这样的:
视频地址
- 首先将双端队列填满到k个。
填充的方式如下:
如果后加入的nums[i]
比双端队列中的最后加入的下标对应的值大,那么将最后加入的下标移出双端队列。如此循环,直到双端队列中目前最后加入的下标对应的值比nums[i]
大,将nums[i]
的下标i
放入双端队列末尾。 - 将双端队列头部的下标对应的值输出到
int[] ans
(也就是题目要求的输出数组)中,并且如果该下标是目前滑动窗口范围内的第一个索引,那么将头部删除,否则什么都不做。 - 滑动窗口范围右移。右移必然引入一个新的
nums[i]
,填充方式与步骤1填充的方式一样。 - 循环步骤2和步骤3,直到滑动窗口到达最后一个数。
直接上代码:
public class Solution {
public int[] maxSlidingWindow(int[] nums, int k) {
ArrayDeque<Integer> arrayDeque=new ArrayDeque();
int[] ans=new int[nums.length-k+1];
int index=0;
for(int i=0;i<nums.length;i++){
if(i>=k){
//将窗口内第一大的数输出
ans[index]=nums[arrayDeque.peekFirst()];
//判断是否需要移除
if (index==arrayDeque.peekFirst()) arrayDeque.removeFirst();
//滑动窗口右移
index++;
}
//填充滑动窗口
while(!arrayDeque.isEmpty()&&nums[arrayDeque.peekLast()]<nums[i]){
arrayDeque.removeLast();
}
if(arrayDeque.size()<k)
arrayDeque.add(i);
}
//再加入最后一个元素
ans[index]=nums[arrayDeque.peekFirst()];
return ans;
}
}