LeetCode 热题100-11 滑动窗口的最大值

万雅过往2024-09-05 8:19

滑动窗口的最大值

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

这个题目也是不好做啊！学了一手单调队列，这个题目的话主要就是让队头存储滑动窗口的最大值，然后后面的要加进来的元素需要比较queue中已经存在元素的大小然后尽快清除

python 复制代码

class Solution:
    def maxSlidingWindow(self, nums: List[int], k: int) -> List[int]:
        queue = []
        for i in range(k):
            while queue and queue[-1]<nums[i]: # 这里还是必须严格大于
                queue.pop(-1)
            queue.append(nums[i])
        ans=[queue[0]]
        for i in range(k,len(nums)):
            if nums[i-k] == queue[0]:queue.pop(0)
            while queue and nums[i]>queue[-1]: # 这里还是必须严格大于
                queue.pop(-1)
            queue.append(nums[i])
            ans.append(queue[0])
        return ans

然后还写了一下窗口未生成和已经生成的版本！简洁一点：

python 复制代码

class Solution:
    def maxSlidingWindow(self, nums: List[int], k: int) -> List[int]:
        queue = []
        res = []
        for i in range(len(nums)):
            if i-k>=0 and queue[0] == nums[i-k]:
                queue.pop(0)
            while queue and queue[-1] < nums[i]:
                queue.pop()
            queue.append(nums[i])
            if i>=k-1:
                res.append(queue[0])
        return res