[路飞]_leetcode刷题_239. 滑动窗口最大值

103 阅读2分钟

题目

239. 滑动窗口最大值

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

解法

思路

单调递减双端队列。逻辑见代码注释

由于该队列需要从头弹出,从尾弹出,从尾推入,而且还需要递减,所以是单调递减双端队列。

代码如下

/**
 * @param {number[]} nums
 * @param {number} k
 * @return {number[]}
 */
var maxSlidingWindow = function(nums, k) {
        if(nums == null || nums.length == 1) return nums;
        // 申明一个单调递减的双向队列,用于存储数据的下标
        let queue = [];
        // 结果数组
        let result = [];
        // 遍历nums数组
        for(let i = 0;i < nums.length;i++){
            // 比较当前元素nums[i]和队列的最后一项last,如果nums[i]>=nums[i],则弹出队列最后一项
            // 循环比较,直到nums[i]小于最后一项为止
            // 正常来说,我们应该推入一个数,并排序,然后弹出最小的一个数
            // 但是由于我们要取最大的数,如果队列的数小于当前元素的话,这些数肯定不会是结果,在队列里还占着位置,直接弹出,时间复杂度还低。
            while(queue.length>0 && nums[queue[queue.length-1]] <= nums[i]){
                queue.pop();
            }
            // 然后将当前元素的下标推入队列
            queue.push(i);
            // 由于窗口每次会向右移动一格,我们需要判断我们队首最大的元素是否还在窗口内
            // 如果不在则要弹出队首元素
            if(queue[0] <= i-k){
                queue.shift();   
            } 
            // 按我们这个遍历的模式,开始窗口长度为0,以此增加,那么只有当窗口长度等于k以后,我们才开始取队首的最大值当结果
            if(i+1 >= k){
                result.push(nums[queue[0]]);
            }
        }
        return result;
};