有一个整形数组arr和一个大小为w的窗口从数组的最左边滑到最右边,窗口每次向右边滑一个位置。
例如,数组为【3,4,1,3,5,3,6】,窗口大小为3时:
[3,4,1[,3,5,3,6 窗口中最大值为4
3,[4,1,3[,5,3,6 窗口中最大值为4
3,4,[1,3,5],3,6 窗口中最大值为5
3,4,1,[3,5,3],6 窗口中最大值为5
3,4,1,3,[5,3,6] 窗口中最大值为6
如果数组长度为n,窗口大小为w,则一共产生n-w+1个窗口的最大值。
上例中,结果应该返回{4,4,5,5,6}
Java代码实现
/*生成最大值数组*/
public class GetMaxWindow {
public static void main(String[] args) {
int[] arr = new int[]{3,4,1,3,5,3,6,4,5,8,9};
arr = getMaxWindwo(arr,3);
for (int value : arr){
System.out.print(value + " ");
}
}
private static int[] getMaxWindwo(int[] arr,int w){
if (arr == null || w < 1 || arr.length < w){
return null;
}
//记录最大值的下标的队列,大小保持小于等于窗口w的大小
LinkedList<Integer> qmax = new LinkedList<>();
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){
//判断队头,如果下标已过期,即不在窗口中了,则应该弹出队头,保证队列的长度小于等于窗口,例如:4321,如果不淘汰则会出现问题。
qmax.pollFirst();
}
if (i >= w -1){
//从第一个窗口的最后的一个下表开始记录,使用peekFirst来取值,而不弹出值
res[index++] = arr[qmax.peekFirst()];
}
}
return res;
}
}