您的位置:首页 > 职场人生

程序员代码面试指南--生成窗口最大值数组

2018-02-10 16:10 597 查看
/**
生成窗口最大值数组
【题目】
有一个整城数组arr和一个大小为w的窗口从数组的最左边滑到最右边,容口每次向右边滑一个位置。
如果数组长度为n,窗口大小为w,则一共产生n-w+1个窗口的最大值。
请实现一个函数。
-输入:整型数组arr,窗口大小为w。
-输出:一个长度为n-w+1的数组res,res[]表示每一种窗口状态下的最大值。
*/
/**
假设遍历到arr[i],qmax的放入规则为:
1.如果qmax为空,直接把下标i敞进qmax,放入过程结束。
2.如果qmax不为空,取出当前qmax队尾存放的下标,假设为j
1)如果arr[j]>arr[i],直接把下标故进qmax的队尾,放入过程结束。
2)如果arr[j]<=arr[i],把j从qmax中弹出,继续qmax的放入规则。
假设遍历到arr[i],qmax的弹出规则为:
如果qmax队头的下标等于i-w,说明当前qmax队头的下标已过期,弹出当前队头的下标即可.
根据如上的放入和弹出规则,qmax便成了一个维护窗口为w的子数组的最大值更新的结构。
*/

import java.util.LinkedList;

public class C07_SlidingWindowMaxArray {
public static void main(String[] args) {
int[] arr = new int[]{4,3,5,4,3,3,6,7};
int w = 3;
printArr(getMaxWindow(arr, w));
}
public static int[] getMaxWindow(int[]arr,int w){
//防护
if(arr==null || w<1 || arr.length< w){
return null;
}
int[]res = new int[arr.length-w+1];
int index = 0;//当前位置的指针
LinkedList<Integer>qmax = new LinkedList<Integer>();//存放下标
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的时候才将最大值加进数组res
res[index++] = arr[qmax.peekFirst()];
}
}
return res;
}
public static void printArr(int[]arr){
for (int i : arr) {
System.out.print(i+" ");
}
System.out.println();
}
}
内容来自用户分享和网络整理,不保证内容的准确性,如有侵权内容,可联系管理员处理 点击这里给我发消息
标签: