0239：滑动窗口最大值（★★）

力扣第 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]

提示：

• 1 <= nums.length <= 105
• -104 <= nums[i] <= 104
• 1 <= k <= nums.length

相似问题：

• 0076：最小覆盖子串
• 0155：最小栈
• 0159：至多包含两个不同字符的最长子串
• 0265：粉刷房子 II
• 1696：跳跃游戏 VI（1954 分）
• 2398：预算内的最多机器人数目（1917 分）
• 2517：礼盒的最大甜蜜度（2020 分）
• 2530：执行 K 次操作后的最大分数（1386 分）

分析

可以直接用有序集合 SortedList 维护有序窗口，不过有更优时间的单调队列解法

• 当窗口内存在 i<j 且 nums[i]<=nums[j] 时，去掉 nums[i] 不影响结果
• 去掉所有这种情况的元素，剩下的即是一个单调递减队列
• 窗口滑动时，从队首弹出过期元素，窗口内的队首即是最大值

解答

 1
 2
 3
 4
 5
 6
 7
 8
 9
10
11
12
13

class Solution:
    def maxSlidingWindow(self, nums: List[int], k: int) -> List[int]:
        res = []
        Q = deque()
        for j,x in enumerate(nums):
            if Q and Q[0][0]==j-k:
                Q.popleft()
            while Q and Q[-1][1]<=x:
                Q.pop()
            Q.append((j,x))
            if j>=k-1:
                res.append(Q[0][1])
        return res

265 ms