有一个整型数组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; } }