数据结构 生成窗口最大值数组

程序员代码面试指南（左程云）读书笔记（7）

第一章

生成窗口最大值数组

题目：

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

   例如：数组为【4、3、5、4、3、3、6、7】，窗口大小为3时：

【4、3、5】4、3、3、6、7     窗口最大值为5

4【3、5、4】3、3、6、7         窗口最大值为5

4、3【5、4、3】3、6、7         窗口最大值为5

4、3、5【4、3、3】6、7          窗口最大值为4

4、3、5、4【3、3、6】7          窗口最大值为6

4、3、5、4、3【3、6、7】         窗口最大值为7

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

   请实现一个函数：

           输入：整形数组arr,窗口大小为w

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

解：

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.addLast(i);

         }

         qmax.addLast(i);

         if(qmax.peekFirst()==i-w){

             qmax.pollFirst();

         }

         if(i>=w-1){

             res[index++]=arr[qmax.peekFirst()];

         }

     }

     return res;

}