本来心心念念是要看 左神 怎么讲清楚这道题的完整过程的,但是!直播的录像被剪了很多,导致我只能靠代码来理解 这个思考过程了;
在leetcode上找到这道题,练了一下手
直接贴左神的题目吧(和leetcode一样的):
题目五
生成窗口最大值数组
【题目】
有一个整型数组 arr 和一个大小为 w 的窗口从数组的最左边滑到最右边,窗口每 次向右边滑一个位置。
例如,数组为[4,3,5,4,3,3,6,7],窗口大小为 3 时:
[4 3 5]4 3 3 6 7窗口中最大值为5 4[3 5 4]3 3 6 7窗口中最大值为5
4 3[5 4 3]3 6 7窗口中最大值为5
4 3 5[4 3 3]6 7窗口中最大值为4
4 3 5 4[3 3 6]7窗口中最大值为6
4 3 5 4 3[3 6 7]窗口中最大值为7
如果数组长度为 n,窗口大小为 w,则一共产生 n-w+1 个窗口的最大值。
请实现一个函数。
输入:整型数组 arr,窗口大小为 w。
输出:一个长度为 n-w+1 的数组 res,res[i]表示每一种窗口状态下的最大值。 以本题为例,结果应该返回{5,5,5,4,6,7}。
因为时间复杂度需要O(n),即一遍扫描出结果,那就需要辅助结构了,可以利用双端队列保存扫描到i时的递减序列(保存的是下标);当扫描值arr[i]>=queue.peekLast(),则表明前面较小的值不可能再成为max了(因为有了arr[i]了丫),所以可以将前面的小值去掉了;如果队列中的first值已经超出了窗口大小,则直接删除(判断方式是 queue.peekFirst()==i-w);
大致思路就是这样,自己都觉得有些乱,但是左神的视频被切了(因为自己有强迫症,思路上不清晰,就感觉很不爽,呃呃呃额额)
代码是我leetcode上写的:(k为window size,nums为arr):
public class Solution {
public int[] maxSlidingWindow(int[] nums, int k) {
if(nums==null || k<1 || nums.length<k){
return new int[0];
}
int[] res=new int[nums.length-k+1];
int index=0;
LinkedList<Integer> maxIndex=new LinkedList<Integer>();
for(int i=0;i<nums.length;i++){
while(!maxIndex.isEmpty() && nums[i]>=nums[maxIndex.peekLast()]){
maxIndex.pollLast();
}
maxIndex.addLast(i);
if(maxIndex.peekFirst()==i-k){
maxIndex.pollFirst();
}
if(i>=k-1){
res[index++]=nums[maxIndex.peekFirst()];
}
}
return res;
}
}