239.滑动窗口的最大值

Posted by 令德湖周杰伦 on 07-28,2020

题目

给定一个数组 nums,有一个大小为 k 的滑动窗口从数组的最左侧移动到数组的最右侧。你只可以看到在滑动窗口内的 k 个数字。滑动窗口每次只向右移动一位。
返回滑动窗口中的最大值。

示例

输入: nums = [1,3,-1,-3,5,3,6,7], 和 k = 3
输出: [3,3,5,5,6,7]
解释:
滑动窗口的位置 | 最大值
[1 3 -1] -3 5 3 6 7 |3
1 [3 -1 -3] 5 3 6 7 |3
1 3 [-1 -3 5] 3 6 7 |5
1 3 -1 [-3 5 3] 6 7 |5
1 3 -1 -3 [5 3 6] 7 |6
1 3 -1 -3 5 [3 6 7] |7

分析

双端队列

  1. 双端队列,长度为k,用来保存窗口内数据,其中windows中保存的是元素下标
  2. 依次加入堆,如果size超过k,就把堆内最小的元素,剔除。
  3. 记录在结果数组中,并返回
public int[] maxSlidingWindowByDequeue(int[] nums, int k) {
        ArrayDeque<Integer> windows = new ArrayDeque<>(k);
        List<Integer> result = new ArrayList<>();
        for (int i = 0; i < nums.length; i++) {
            int x = nums[i];
            //淘汰掉前面的老元素
            if(i >= k && windows.getFirst() <= i - k){
                windows.pop();
            }
            //如果窗口内有小于当前元素的老元素,那么全部剔除, 从后开始比较,并且删除
            while(!windows.isEmpty() && nums[windows.getLast()]<= x){
                windows.removeLast();
            }
            //加入队列
            windows.add(i);
            //当窗口移动到k时才往结果集放元素
            if(i>=k-1){
                result.add(nums[windows.getFirst()]);
            }
        }
        int a[] = new int[result.size()];
        for (int i = 0; i < result.size(); i++) {
            a[i] = result.get(i);
        }
        return a;
    }