239. 滑动窗口最大值

239. 滑动窗口最大值

原始题目链接:https://leetcode.cn/problems/sliding-window-maximum/

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

解题思路:

使用单调队列来解决滑动窗口的问题。求最大值,为了方便,保持队列的单调递减的特性,这样队头就是最大值。滑动窗口满了就移出最开始入队的元素,即队头元素。新元素入队,如果新元素比队里的元素(是当前元素前面的元素,即遍历过的元素且在队列中)都大,那么队列中的元素不可能成为最大值,弹出队列中的元素即可。

代码实现:

class Solution:
    def maxSlidingWindow(self, nums: List[int], k: int) -> List[int]:
        if not nums or k == 0:
            return []
        
        if len(nums) == 1:
            return [nums[0]]

        # 维护一个单调递减的队列,保存的是滑动窗口的值
        # 队头元素是最大值,队列里存储的是元素在数组中的下标
        queue = collections.deque()
        ans = []

        # 遍历nums
        for i in range(len(nums)):
            # 滑动窗口中的元素满了,即元素大小等于k个
            # 滑动窗口有k个元素:队头元素的下标queue[0] 和当i-k相等
            # 为了能让下个元素进入滑动窗口,需要弹出队头元素,队头元素是最先进入队列的元素
            if queue and queue[0] == i - k:
                queue.popleft()
            
            # 保持单调递减的特性
            # 如果新进入的元素比队列里的元素(必须是当前元素前面的元素)大,那么队列中的元素不可能是接下来滑动窗口的最大值
            # 可以将小于当前元素的值出队,再将新元素入队,为了保持递减的特性,从最小的元素开始比较,
            # 所以出队的顺序是从右边出,因为最右边的元素最小
            while queue and nums[queue[-1]] < nums[i]:
                queue.pop()
            
            # 新元素入队
            queue.append(i)

            # 获取滑动窗口的最大值
            # 当前元素的下标和k相差大于等于1,表示滑动窗口已经有k个值
            if i >= k - 1:
                ans.append(nums[queue[0]])
        return ans

参考文献:
https://leetcode.cn/problems/sliding-window-maximum/solution/acm-xuan-shou-tu-jie-leetcode-hua-dong-c-i3wj/

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值