239.滑动窗口的最大值

小小皮卡丘啃算法2024-04-14 13:01

·题目描述

复制代码

给你一个整数数组 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.创建一个非增双端队列（使用collection），用来存放最大值

2.循环窗口的长度为k，也就是两个指标，i，和j（i表示开端，就表示末端）。用来遍历数组

3.判断条件。判断条件有三个：

---滑动窗口，如果左端滑出的值为q队列中的最大值，那么需要在滑动后将队列左端的数值弹出

---如果队列存在，并且窗口的值大于队列末端值，将队列的末端数值弹出。再添加数值到队列中

---当窗口左端的指标从0 开始的时候，就可以向结果数组中添加q队列的初始值。

·代码

python 复制代码

import collections
class Solution:
    def maxSlidingWindow(self, nums, k: int):
        n = len(nums)
        q = collections.deque()
        result = set()

        for i, j in zip(range(1-k,n-k+1),range(0,n)):
            if i > 0 and nums[i-1] == q[0]:
                q.popleft()

            while q and nums[j]> q[-1]:
                q.pop()
            q.append(nums[j])

            if i >= 0:
                result.add(q[0])

        return result