滑动窗口最大值数组

2017-10-29 23:40:30

题目描述:

有一个整型数组arr和一个大小为w的窗口从数组的最左边滑到最右边,窗口每次向右边滑一个位置。窗口中每次会产生一个最大值,求这n-w+1最大值组成的数组。

要求时间复杂度为O(n)。

求解:

举例:[4,3,5,4,3,3,6,7],窗口大小为3

结果:[5,5,5,4,6,7]

采用一个双端队列来维护窗口中的最大值,以及可能成为最大值的数。

定义一个双端队列qmax用来保存最大值的下标。

当遍历到arr[i]的时候,

~ 进队列的策略:

1、如果qmax为空,则直接将下标放入队列

2、如果qmax非空,则比较arr[i]和队列中的队尾j对应的arr[j]的大小。

           若arr[j]>arr[i],那么表示arr[i]有可能在接下来成为最大值,所以将i放入队列

           若arr[j]<=arr[i],那么表示a[j]已经不可能是最大的值了,所以将arr[j]抛出,继续进行比较。

~ 出队列的策略:

如果此时队首的下标k满足i-k=w,则表示当前的队首元素已经过期,所以可以将队首的元素抛出。

public class MaxWindow {
    public int[] getMaxWindow(int[] arr,int w)
    {
        LinkedList<Integer> qmax = new LinkedList<>();
        int[] rst=new int[arr.length-w+1];
        int j=0;
        for(int i=0;i<arr.length;++i){
            while(!qmax.isEmpty()&&(arr[qmax.getLast()]<=arr[i])) qmax.removeLast();
            qmax.add(i);
            if(!qmax.isEmpty()&&(qmax.getFirst()==i-w)) qmax.removeFirst();
            if(i>=w-1) rst[j++]=arr[qmax.getFirst()];
        }
        return rst;
    }
}

 

posted @ 2017-10-31 13:58  hyserendipity  阅读(763)  评论(0编辑  收藏  举报