leetcode 239 滑动窗口题解

118 阅读2分钟

题目描述

给你一个整数数组 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.首先建立一个数组ans存储最后的答案,建立一个双端队列deque用来存储数据的位置,且位置对应的数据是单调递减的,位置下标是单调递增的。
2.先遍历第一个窗口,每遍历一个值,在将他的位置插入前,从队尾开始判断,小于等于它的值的位置均弹出,最后得到一个初始的deque,队首就是第一个窗口的最大值的位置,将其对应的值插入到ans。
3.假设一个窗口的两边指针是start、end,则遍历下一个窗口时,起始位置改成start+1,所以如果start位置存在于队列,那他一定是在队首,这是要将start位置弹出。然后在插入end+1位置前,也要先弹出前方比他小的数的位置。此时队首位置对应的数据就是当前窗口最大的值的位置。

过程

nums = [1,3,-1,-3,5,3,6,7], k = 3

1644566274(1).jpg 1644566358(1).jpg

代码

var maxSlidingWindow = function(nums, k) {
    const deque = []
    const ans = []

    for(let i = 0; i < k; i++) {
        while(deque.length > 0 && nums[i] >= nums[deque[deque.length-1]]) deque.pop()

        deque.push(i)
    }

    ans.push(nums[deque[0]])
    for(let i = k; i < nums.length; i++) {
        while(deque.length > 0 && nums[i] >= nums[deque[deque.length-1]]) deque.pop()

        if(deque.length > 0 && i-k >= deque[0]) deque.shift()

        deque.push(i)
        ans.push(nums[deque[0]])
    }

    return ans
};