九章算法 | 微软面试题:滑动窗口的最大值

给出一个可能包含重复的整数数组,和一个大小为 k 的滑动窗口, 从左到右在数组中滑动这个窗口,找到数组中每个窗口内的最大值。

在线评测地址:LintCode 领扣

样例 1:

输入:
[1,2,7,7,8]
3
输出:
[7,7,8]

解释:
最开始,窗口的状态如下:`[|1, 2 ,7| ,7 , 8]`, 最大值为 `7`;
然后窗口向右移动一位:`[1, |2, 7, 7|, 8]`, 最大值为 `7`;
最后窗口再向右移动一位:`[1, 2, |7, 7, 8|]`, 最大值为 `8`. 

样例 2:

输入:
[1,2,3,1,2,3]
5
输出:
[3,3]

解释:
最开始,窗口的状态如下: `[|1,2,3,1,2 | ,3]` , 最大值为`3`;
然后窗口向右移动一位.`[1, |2,3,1,2,3]`, 最大值为 `3`; 

算法:单调队列

  • 单调队列中的元素是严格单调的。我们在求解这个问题的时候需要维护他的单调性。
  • 队首元素即为当前位置的最大值。假设要求滑动窗口中的最大值。我们就需要确保滑动窗口中的元素从队首到队尾是递减的。
  • 每滑动一次就判断当前元素和队尾元素的关系,如果放入队尾满足单调递减,那么放入即可;如果放入不满足,就需要删除队尾元素直到放入当前元素之后满足队列单调递减。同时要确保已经出窗口的最大值(队首元素)被删除掉。

注意事项:

我们在本题中 往单调队列里压入的是元素对应的下标,因为我们要通过下标来判断该元素有没有离开了滑动窗口,如果离开了滑动窗口就得从队列里给删除了

流程

序列a是 1 2 7 7 6 5 3

滑动窗口的大小是3

遍历一遍 for i in 0:7

  • i==0 队列里没有元素,直接添加(value=1,index=0) value是权值,index是下标(这里为了表达清晰,就把权值和下标都注明出来了)

队列里 (1,0)

  • i==1 a[1]比队尾元素大,添加a[1]后不满足单调递减性质,所以需要把队尾给弹出,a[1]比队尾大,而且队尾元素肯定在a[1]的左面(因为队尾元素先压入的,它的下标肯定小),而且队尾元素还没a[1]大,对于后面的窗口,队尾元素肯定不是最大值了(我们已经找到了 a[1]比,而且队尾在窗口里面,a[1]肯定也在窗口里面,因为a[1]的下标更大)
    • 弹掉队尾后,队列为空
    • 压入(2,1)
    • 队列里 (2,1)
  • i==2 a[2]比队尾元素大,弹掉队尾,压入(7,2)

队列里 (7,2)

这里i>=k-1了 ,当前队首为当前[0,2]这个滑动窗口的答案,这个窗口答案是7

  • i==3 a[3]不比队尾元素大,不用弹出队尾 ,压入新元素

队列里 (7,2) (7,3)

统计[1,3]滑动窗口的答案,也是7

  • i==4 a[4]不比队尾元素大,不用弹出队尾,压入新元素

队列里 (7,2) (7,3) (6,4)

统计[2,4]滑动窗口的答案,也是7

  • i==5 a[5]不比队尾元素大,不用弹出队尾,压入新元素

队列里 (7,2) (7,3)(6,4)(5,5)

这时候i和队首位置差已经超过了滑动窗口长度,弹出队首

队列里 (7,3) (6,4)(5,5)

这个窗口答案也为7

复杂度分析

时间复杂度:每个元素进队出队都只有一次,所以复杂度是O(N)的

空间复杂度:队列里元素最多为N个,所以空间复杂度也是O(N)

public class Solution {
    /**
     * @param nums: A list of integers.
     * @param k: An integer
     * @return: The maximum number inside the window at each moving.
     */
    public List<Integer> maxSlidingWindow(int[] nums, int k) {
        // write your code here
        // 答案数组和单调队列
        List<Integer> ans = new ArrayList<Integer>();
        Deque<Integer> qmax = new LinkedList<Integer>();
        int n = nums.length;
        for(int i = 0; i < n; i++) {
            //维护单调性
            while(!qmax.isEmpty() && nums[i] > nums[qmax.getLast()])
                qmax.removeLast();
            //滑动窗口的长度超过了k 删掉超过的部分
            if(!qmax.isEmpty() && i - qmax.getFirst() >= k)
                qmax.removeFirst();

            qmax.addLast(i);

            //统计答案
            if(i >= k - 1) {
                ans.add(nums[qmax.getFirst()]);
            }
        }
        return ans;
    }
}

更多题解参考:九章算法

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值