滑动窗口的最大值

题目

有一个整型数组 arr 和一个大小为 w 的窗口从数组的最左边滑到最右边,窗口每次向右边滑一个位置。 返回一个长度为n-w+1的数组res,res[i]表示每一种窗口状态下的最大值。 以数组为[4,3,5,4,3,3,6,7],w=3为例。因为第一个窗口[4,3,5]的最大值为5,第二个窗口[3,5,4]的最大值为5,第三个窗口[5,4,3]的最大值为5。第四个窗口[4,3,3]的最大值为4。第五个窗口[3,3,6]的最大值为6。第六个窗口[3,6,7]的最大值为7。所以最终返回[5,5,5,4,6,7]。
给定整形数组arr及它的大小n,同时给定w,请返回res数组。保证w小于等于n,同时保证数组大小小于等于500。

测试样例:[4,3,5,4,3,3,6,7],8,3
返回:[5,5,5,4,6,7]

思路

用双端队列deque保存可能成为滑动窗口中最大元素的下标,进出队列规则如下:

  1. 如果deque为空,直接将下标i放入deque中。
  2. 如果当前下标i-队头元素==w,弹出队头元素。
  3. 如果deque不为空,取出deque队尾的下标j,如果arr[j]>=arr[i],将下标i放入deque队尾。否则依次弹出队尾元素,直到arr[j]<arr[i]或者deque为空,再将i压入队尾。

1和2都比较好理解,这里解释一下3:

如果arr[j]>=arr[i],将下标i放入deque队尾。否则依次弹出队尾元素,直到arr[j]<arr[i]或者deque为空,再将i压入队尾.

arr[j]>=arr[i],为什么要放在队尾呢,arr[i]虽然此时比arr[j]小,但是arr[j]会在一定时间后失效(即脱出滑动窗口范围),arr[i]可能会成为最大值。
而当arr[j]<arr[i],说明arr[j]永远不能成为当前窗口及以后窗口的最大值了,因为arr[j]不仅比arr[i]小,而且一定比arr[j]先失效。所以将其弹出即可。

当初始队列构造完成后,依次读入数组元素,队列的头元素的下标所代表的数字就是此时窗口中的最大值。

原文:https://www.jianshu.com/p/3242ddf81428


原创部分:

原文的解释是很到位的,我再补充一点儿,算法复杂度为O(n),可能乍一看,会觉得这个复杂度是O(nw)因为规则3有一个向前遍历的过程。其实不然,我们可以这样考虑,对于数组中的每个元素来说,最多入队,出队一次,所以这个时间复杂度就是O(n)

以下是我给出的java代码实现:

    public ArrayList<Integer> maxInWindows(int [] num, int size)
    {
        if(size <= 0) 
            return new ArrayList<Integer>();
        
        Deque<Integer> dq = new LinkedList<>();
        ArrayList<Integer> al = new ArrayList<>();
        for(int i=0; i<num.length; i++){
            if(dq.isEmpty())            //规则1
                dq.addFirst(i);
            else{                        //规则3
                while(!dq.isEmpty()){
                    if(num[dq.getLast()] < num[i])
                        dq.removeLast();
                    else break;
                }
                dq.addLast(i);
            }
            if(i-dq.getFirst() == size)    //规则2
                dq.removeFirst();
            
            if(i>=size-1)                //当凑够了一个窗口才开始添加最大值
                al.add(num[dq.getFirst()]);
        }
        return al;
        
    }

从这里我们也可以看出,代码固然重要,但是更重要的是思想。不过也要有扎实的基本功,否则你连Deque怎么用都不知道,那就没办法把思路转换成代码,用代码实现思路这个过程需要大量的积累和练习。

 

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值