题目描述:给定一个数组和滑动窗口k的大小,找出所有滑动窗口里数值的最大值。例如,如果输入数组{2,3,4,2,6,2,5,1}及滑动窗口的大小3,那么一共存在6个滑动窗口,他们的最大值分别为{4,4,6,6,6,5}; 针对数组{2,3,4,2,6,2,5,1}的滑动窗口有以下6个: {[2,3,4],2,6,2,5,1}, {2,[3,4,2],6,2,5,1}, {2,3,[4,2,6],2,5,1}, {2,3,4,[2,6,2],5,1}, {2,3,4,2,[6,2,5],1}, {2,3,4,2,6,[2,5,1]}。分别找出这6个滑动窗口中的最大值,并保存到一个列表中。
最简单的思路就是分别找到这些滑动窗口,并用一个数组来保存,然后找出每个数组的最大值,但是因为比较每个数组需要时间复杂度为O(k),滑动窗口滑动的时间复杂度为O(n),所以总的时间复杂度为O(nk);
那么我们希望在窗口滑动的同时,进行最大值的更新,那么可以使用一个双向的队列来保存滑动窗口中的值,使最大值总是在双向队列的头部,新的元素和对尾的元素比较,如果队尾的元素小于新的元素,就把队尾的元素删除,然后从队尾加入新的元素。但是怎么判断在队列头的最大值是否过期呢,那么这就需要在双向队列中保存的是元素的下标,当发现队列头部的元素下标和当前下标的差值大于滑动窗口的大小的时候,说明该最大值过期,应该把他从队列头部删除掉。
实现的代码如下:
public class Solution {
public ArrayList<Integer> maxInWindows(int [] num, int size)
{
//这道题的解题思路主要有两点:
//1、用一个双端队列ArrayDeque来保存滑动窗口中的元素
//2、双端队列里面保存的是元素的下标的值,而不是元素本身的值,这点非常重要,用于判断元素是否过期
ArrayList<Integer> aList = new ArrayList<>();
if(num == null || size<1 || num.length<size){
return aList;
}
ArrayDeque<Integer> aDeque = new ArrayDeque<>();
int begin;
for(int i = 0; i<num.length; i++){
begin = i-size+1;
if (aDeque.isEmpty()){
aDeque.add(i);
}
else if(begin>aDeque.peekFirst()){
aDeque.pollFirst();
}
while(!aDeque.isEmpty() && num[i]>=num[aDeque.peekLast()]){
aDeque.pollLast();
}
aDeque.add(i);
if(begin >= 0){
aList.add(num[aDeque.peekFirst()]);
}
}
return aList;
}
}