239. 滑动窗口最大值

心已赠人 2023-02-08 12:44 20阅读 0赞

题目链接：https://leetcode-cn.com/problems/sliding-window-maximum/

> 给你一个整数数组 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

思路：使用双端队列来表示窗口，双端队列就是队首队尾都可以增加或者减少元素。  
本题要点在于保证队首元素永远是当前队列的最大值，这样便于取第一个元素放到结果集。  
思考两个问题：  
1 用什么数据结构表示滑动窗口？  
答：本题解使用双端队列。

2 怎么取出窗口的最大值？  
答：队首元素就是最大值

3 什么时候可以取出窗口最大值放到结果集中？  
答：下标 i 比窗口 k 大之后

代码

class Solution {
        
        public int[] maxSlidingWindow(int[] nums, int k) {
        
    
            int[] result = new int[nums.length-k+1];
            // 用最常见的List来表示窗口（双端队列）
            ArrayList<Integer> window = new ArrayList<>();
    
            for (int i = 0 ; i < nums.length; i++) {
        
    
                while (!window.isEmpty() && nums[window.get(window.size() - 1)] <= nums[i] ) {
        
                    window.remove(window.size() - 1);
                }    
    
                window.add(i);
    
                // 窗口大小是3，需要把已经小于范围的下标去掉
                if (window.get(0) <= i - k) {
        
                    window.remove(0);
                }
    
                // 什么时候可以把结果放到结果集？答：下标比窗口大的时候
                if (i + 1 >= k) {
        
                    // 放到哪个位置？答：每次遍历都要放，顺序放到结果集的位置
                    result[i + 1 - k] = nums[window.get(0)];
                }
    
    
            }
            return result;
        }
    }

推荐题解：  
https://leetcode-cn.com/problems/sliding-window-maximum/solution/shuang-xiang-dui-lie-jie-jue-hua-dong-chuang-kou-2/