算法总结之 生成窗口的最大值数组

有一个整型数组arr 和一个 大小为w的窗口从数组的最左边滑倒最右边，窗口每次向右边滑动一个位置

如果数组长度为n， 窗口大小为w， 则一共产生 n-w+1 个窗口的最大值

 

请实现一个函数：

 输入  整型数组 arr， 窗口大小 w

 输出 一个长度为 n-w+1 的数组res  res[i] 表示每一种窗口状态下的最大值。

 

 

本题的关键在于利用双端队列来实现最大值的更新。首先生成双端队列qmax， qmax中存放数组arr的下标

 

看代码：

package TT;

import java.util.LinkedList;

public class Test125 {

	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){
				res[index++]=arr[qmax.peekFirst()];
			}
		}
		return res;
	}
	
	
	
}