题目:有一个整型数组arr和一个大小为w的窗口从数组的最左边滑到最右边,窗口每次向右边滑一个位置。如果数组长度为n,窗口大小为w,则一共产生n - w + 1 个窗口的最大值。
分析:
利用双端队列来实现窗口最大值的更新,从头端删除,从尾端插入和删除,属于(输入受限的双端队列),每个下标最多进队列一次,出队列一次,所以遍历过程中进出双端队列的操作的时间复杂度为O(N),整体的时间复杂度也为O(N)。首先生成双端队列qmax,队列中存放数组arr中的下标。
假设遍历到 arr[i] , qmax的放入规则为:
1:如果 qmax 为空, 直接放入i,放入过程结束。
2:如果qmax不为空, 取出当前队列的队尾存放的下标,假设为j,
a)如果 arr[j] > arr[i] ,直接把下标i插入队尾,放入结束。
b)如果arr[j] <= arr[i] , 把 j 从 队尾 弹出, 继续 放入规则。
加入遍历到 arr[i], qmax的弹出规则为:
如果队头的下标等于 i - w ,说明当前队头的下标已经过期,弹出队头下标即可。
代码如下:
import java.util.*;
public class Main {
public static void main(String[]args){
Main test = new Main() ;
int []arr = {4 ,3 ,5 ,4 ,3 ,3 ,6 ,7} ;
int w = 3 ;
int []res = test.getMaxWindow(arr, w) ;
System.out.println(Arrays.toString(res));
}
public int[] getMaxWindow(int [] arr , int w){
if(arr == null || w < 1 || arr.length < w){
return null ;
}
LinkedList<Integer> qmax = new LinkedList<Integer>() ;
int [] res = new int[arr.length - w + 1] ;
int index = 0 ;
for(int i = 0 ; i<arr.length ; i++){
while(!qmax.isEmpty() && arr[qmax.peekLast()] <= arr[i]){
qmax.pollLast() ; //队尾弹出较小值的下标
}
qmax.addLast(i); //队尾放入下标
if(qmax.peekFirst() == i - w){ // 队头弹出过期的下标
qmax.pollFirst();
}
if(i >= w - 1){ //从 w -1 位开始保存窗口最大值
res[index++] = arr[qmax.peekFirst()] ;
}
}
return res ;
}
}