问题

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

返回滑动窗口中的最大值

示例 1:
输入: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

示例 2:
输入:nums = [1], k = 1
输出:[1]

示例 3:
输入:nums = [1,-1], k = 1
输出:[1,-1]

示例 4:
输入:nums = [9,11], k = 2
输出:[11]

示例 5:
输入:nums = [4,-2], k = 2
输出:[4]

思路

对于每个滑动窗口,我们可以使用 leetcode-239:滑动窗口最大值 - 图1 的时间遍历其中的每一个元素,找出其中的最大值。对于长度为 n 的数组 nums 而言,窗口的数量为 n-k+1,因此该算法的时间复杂度为leetcode-239:滑动窗口最大值 - 图2,会超出时间限制,因此我们需要进行一些优化

我们可以想到,对于两个相邻(只差了一个位置)的滑动窗口,它们共用着 k-1 个元素,而只有 1 个元素是变化的。我们可以根据这个特点进行优化

方法一:优先队列

对于最大值,可以想到一种非常合适的数据结构,那就是优先队列(堆),其中的大根堆可以帮助我们实时维护一系列元素中的最大值

对于本题而言,初始时,我们将数组 nums 的前 k 个元素放入优先队列中。每当我们向右移动窗口时,我们就可以把一个新的元素放入优先队列中,此时堆顶的元素就是堆中所有元素的最大值。然而这个最大值可能并不在滑动窗口中,在这种情况下,这个值在数组 nums 中的位置出现在滑动窗口左边界的左侧。因此,当我们后续继续向右移动窗口时,这个值就永远不可能出现在滑动窗口中了,我们可以将其永久地从优先队列中移除

我们不断地移除堆顶的元素,直到其确实出现在滑动窗口中。此时,堆顶元素就是滑动窗口中的最大值。为了方便判断堆顶元素与滑动窗口的位置关系,我们可以在优先队列中存储二元组 (num,index),表示元素 num 在数组中的下标为 index
leetcode-239:滑动窗口最大值 - 图3
leetcode-239:滑动窗口最大值 - 图4
leetcode-239:滑动窗口最大值 - 图5
leetcode-239:滑动窗口最大值 - 图6
leetcode-239:滑动窗口最大值 - 图7
leetcode-239:滑动窗口最大值 - 图8
在Java中的优先队列(PriorityQueue)就是堆的数据结构

  1. class Solution {
  2. public int[] maxSlidingWindow(int[] nums, int k) {
  3. int n = nums.length;
  4. // 1. 优先队列存放的是二元组(num,index) : 大顶堆(元素大小不同按元素大小排列,元素大小相同按下标进行排列)
  5. // num : 是为了比较元素大小
  6. // index : 是为了判断窗口的大小是否超出范围
  7. // 当两者的值相同时,比较下标的位置,下标大的在前面
  8. PriorityQueue<int[]> pq = new PriorityQueue<int[]>(new Comparator<int[]>(){
  9. public int compare(int[] pair1,int[] pair2){
  10. return pair1[0] != pair2[0] ? pair2[0] - pair1[0]:pair2[1] - pair1[1];
  11. }
  12. });
  13. // 2. 优先队列初始化 : k个元素的堆
  14. for(int i = 0; i < k; i++){
  15. pq.offer(new int[]{nums[i],i});
  16. }
  17. // 3. 处理堆逻辑
  18. int[] res = new int[n - k + 1]; // 初始化结果数组长度 :一共有 n - k + 1个窗口
  19. res[0] = pq.peek()[0]; // 初始化res[0] : 拿出目前堆顶的元素
  20. for(int i = k;i < n;i++){ // 向右移动滑动窗口
  21. pq.offer(new int[]{nums[i],i}); // 加入大顶堆中
  22. while(pq.peek()[1] <= i - k){ // 循环判断当前队首是否在窗口中,窗口的左边界为i-k
  23. pq.poll(); // 维护:堆的大小就是滑动窗口的大小
  24. }
  25. res[i - k + 1] = pq.peek()[0]; // 此时堆顶元素就是滑动窗口的最大值
  26. }
  27. return res;
  28. }
  29. }
  • 时间复杂度:leetcode-239:滑动窗口最大值 - 图9,其中 n 是数组 nums 的长度。在最坏情况下,数组 nums 中的元素单调递增,那么最终优先队列中包含了所有元素,没有元素被移除。由于将一个元素放入优先队列的时间复杂度为 leetcode-239:滑动窗口最大值 - 图10,因此总时间复杂度为 leetcode-239:滑动窗口最大值 - 图11
  • 空间复杂度:leetcode-239:滑动窗口最大值 - 图12,即为优先队列需要使用的空间。这里所有的空间复杂度分析都不考虑返回的答案需要的 leetcode-239:滑动窗口最大值 - 图13 空间,只计算额外的空间使用

方法二:单调队列

我们可以顺着方法一的思路继续进行优化

由于我们需要求出的是滑动窗口的最大值,如果当前的滑动窗口中有两个下标 ij,其中 ij 的左侧(i < j),并且 i 对应的元素不大于 j 对应的元素(nums[i]≤nums[j]),那么会发生什么呢?

当滑动窗口向右移动时,只要 i 还在窗口中,那么 j 一定也还在窗口中,这是 ij 的左侧所保证的。因此,由于 nums[j] 的存在,nums[i] 一定不会是滑动窗口中的最大值了,我们可以将 nums[i] 永久地移除

因此我们可以使用一个队列存储所有还没有被移除的下标。在队列中,这些下标按照从小到大的顺序被存储,并且它们在数组 nums 中对应的值是严格单调递减的。因为如果队列中有两个相邻的下标,它们对应的值相等或者递增,那么令前者为 i,后者为 j,就对应了上面所说的情况,即 nums[i] 会被移除,这就产生了矛盾

当滑动窗口向右移动时,我们需要把一个新的元素放入队列中。为了保持队列的性质,我们会不断地将新的元素与队尾的元素相比较,如果前者大于等于后者,那么队尾的元素就可以被永久地移除,我们将其弹出队列。我们需要不断地进行此项操作,直到队列为空或者新的元素小于队尾的元素

由于队列中下标对应的元素是严格单调递减的,因此此时队首下标对应的元素就是滑动窗口中的最大值。但与方法一中相同的是,此时的最大值可能在滑动窗口左边界的左侧,并且随着窗口向右移动,它永远不可能出现在滑动窗口中了。因此我们还需要不断从队首弹出元素,直到队首元素在窗口中为止

为了可以同时弹出队首和队尾的元素,我们需要使用双端队列。满足这种单调性的双端队列一般称作单调队列(单调队列是在队列中所有的元素都是单调的,要么单调增,要么单调减,新增元素时到队尾,此时队列不符合单调性就将队尾元素出队,直到队列单调或空)

leetcode-239:滑动窗口最大值 - 图14
leetcode-239:滑动窗口最大值 - 图15
leetcode-239:滑动窗口最大值 - 图16
leetcode-239:滑动窗口最大值 - 图17
leetcode-239:滑动窗口最大值 - 图18
leetcode-239:滑动窗口最大值 - 图19
leetcode-239:滑动窗口最大值 - 图20
leetcode-239:滑动窗口最大值 - 图21

  1. public int[] maxSlidingWindow(int[] nums, int k) {
  2. int n = nums.length;
  3. //创建双端队列
  4. Deque<Integer> deque = new ArrayDeque<>();
  5. //先初始化前K个元素
  6. for (int i = 0; i < k; i++) {
  7. //判断队列是否为空 或者当前入队元素是否大于队尾元素 大于则出队
  8. while (!deque.isEmpty() && nums[i] >= nums[deque.peekLast()]) {
  9. deque.pollLast();
  10. }
  11. //当前元素入队
  12. //由于需要判断当前元素是否在窗口中,所以实际上队列中存储的为当前元素的下标
  13. //根据下标找元素比根据元素找下标方便
  14. deque.offerLast(i);
  15. }
  16. int[] ans = new int[n - k + 1];
  17. //添加当前最大元素
  18. ans[0] = nums[deque.peekFirst()];
  19. for (int i = k; i < n; i++) {
  20. //判断队列是否为空 或者当前入队元素是否大于队尾元素 大于则出队
  21. while (!deque.isEmpty() && nums[i] >= nums[deque.peekLast()]) {
  22. deque.pollLast();
  23. }
  24. //当前元素入队
  25. deque.offerLast(i);
  26. //判断队首元素是否在窗口中
  27. while (deque.peekFirst() <= i - k) {
  28. deque.pollFirst();
  29. }
  30. //添加答案
  31. ans[i - k + 1] = nums[deque.peekFirst()];
  32. }
  33. return ans;
  34. }
  • 时间复杂度:leetcode-239:滑动窗口最大值 - 图22,其中 n 是数组 nums 的长度。每一个下标恰好被放入队列一次,并且最多被弹出队列一次
  • 空间复杂度:leetcode-239:滑动窗口最大值 - 图23。与方法一不同的是,在方法二中我们使用的数据结构是双向的,因此「不断从队首弹出元素」保证了队列中最多不会有超过 k+1 个元素

方法三:分块 + 预处理思路与算法

除了基于随着窗口的移动实时维护最大值的方法一以及方法二之外,我们还可以考虑其他有趣的做法

我们可以将数组 nums 从左到右按照 k 个一组进行分组,最后一组中元素的数量可能会不足 k 个。如果我们希望求出 nums[i]nums[i+k−1] 的最大值,就会有两种情况:

  • 如果 ik 的倍数,那么 nums[i]nums[i+k−1] 恰好是一个分组。我们只要预处理出每个分组中的最大值,即可得到答案;
  • 如果 i 不是 k 的倍数,那么 nums[i]nums[i+k−1] 会跨越两个分组,占有第一个分组的后缀以及第二个分组的前缀。假设 jk 的倍数,并且满足 i < j ≤ i+k−1,那么 nums[i]nums[j−1] 就是第一个分组的后缀,nums[j]nums[i+k−1] 就是第二个分组的前缀。如果我们能够预处理出每个分组中的前缀最大值以及后缀最大值,同样可以在 leetcode-239:滑动窗口最大值 - 图24 的时间得到答案

因此我们用 prefixMax[i] 表示下标 i 对应的分组中,以 i 结尾的前缀最大值;suffixMax[i] 表示下标 i 对应的分组中,以 i 开始的后缀最大值。它们分别满足如下的递推式

leetcode-239:滑动窗口最大值 - 图25

以及

leetcode-239:滑动窗口最大值 - 图26

需要注意在递推 suffixMax[i] 时需要考虑到边界条件 suffixMax[n−1]=nums[n−1],而在递推 prefixMax[i] 时的边界条件 prefixMax[0]=nums[0] 恰好包含在递推式的第一种情况中,因此无需特殊考虑

在预处理完成之后,对于 nums[i]nums[i+k−1] 的所有元素,如果 i 不是 k 的倍数,那么窗口中的最大值为 suffixMax[i]prefixMax[i+k−1] 中的较大值;如果 ik 的倍数,那么此时窗口恰好对应一整个分组,suffixMax[i]prefixMax[i+k−1] 都等于分组中的最大值,因此无论窗口属于哪一种情况,

leetcode-239:滑动窗口最大值 - 图27

即为答案

class Solution {
    public int[] maxSlidingWindow(int[] nums, int k) {
        int n = nums.length;
        int[] prefixMax = new int[n];
        int[] suffixMax = new int[n];
        for (int i = 0; i < n; ++i) {
            if (i % k == 0) {
                prefixMax[i] = nums[i];
            }
            else {
                prefixMax[i] = Math.max(prefixMax[i - 1], nums[i]);
            }
        }
        for (int i = n - 1; i >= 0; --i) {
            if (i == n - 1 || (i + 1) % k == 0) {
                suffixMax[i] = nums[i];
            } else {
                suffixMax[i] = Math.max(suffixMax[i + 1], nums[i]);
            }
        }

        int[] ans = new int[n - k + 1];
        for (int i = 0; i <= n - k; ++i) {
            ans[i] = Math.max(suffixMax[i], prefixMax[i + k - 1]);
        }
        return ans;
    }
}
  • 时间复杂度:leetcode-239:滑动窗口最大值 - 图28,其中 n 是数组 nums 的长度。我们需要 leetcode-239:滑动窗口最大值 - 图29 的时间预处理出数组 prefixMaxsuffixMax 以及计算答案
  • 空间复杂度:leetcode-239:滑动窗口最大值 - 图30,即为存储 prefixMaxsuffixMax 需要的空间