题目:有一个整数型数组arr和一个大小为w的窗口从数组的最左滑到最右边,窗口每次向右边滑一个位置(时间复杂度O(n)),如果数组长度为n,窗口大小为w,则一共生产n-w+1个窗口的最大值。
import org.junit.Test;
import java.util.LinkedList;
public class getMaxWindow {
@Test
public void Test(){
int []arr={9,7,6,5,4,8,3};
int [] max=getMaxWindow(arr,3);
for(int i:max){
System.out.println(i);
}
}
/**
*
* @param arr 数组
* @param w 窗口的大小
* @return
*/
public int[] getMaxWindow(int []arr,int w){
if(arr==null || w<1||arr.length<w){
return null;
}
LinkedList<Integer> max=new LinkedList<>();
int []res=new int[arr.length-w+1];//最大值数组
int index=0;
//因为每个下标值最多进队列一次,出队列一次,所以时间复杂度为O(n)
for(int i=0;i<arr.length;i++){
//将队列中的最小值和将要加入的值进行比较,如果小于等于则把队列尾部值poll,然后继续比较
while (!max.isEmpty()&&arr[max.peekLast()]<=arr[i]){
max.pollLast();
}
max.addLast(i);
if(max.peekFirst()==i-w){//是否在窗口大小内
max.pollFirst();
}
if(i>=w-1){//当i大于窗口的大小后,每往右移一位都有一个最大值
res[index++]=arr[max.peekFirst()];//队列头部是最大值,上一个while循环保证
}
}
return res;
}
}
运行结果: