单调队列-滑动窗口最大值

最新推荐文章于 2024-07-16 23:15:32 发布

hema12138

最新推荐文章于 2024-07-16 23:15:32 发布

阅读量357

点赞数 4

文章标签： leetcode 数据结构

本文链接：https://blog.csdn.net/hema12138/article/details/136222094

版权

本文介绍了一个编程问题，如何使用Python实现滑动窗口算法，找出数组中每滑动k个元素时的最大值。通过单调队列结构保持窗口内的最大值，并分析了其时间复杂度O(n)和空间复杂度O(k)。

摘要由CSDN通过智能技术生成

Problem: 239. 滑动窗口最大值

题目

输入一个数组nums，滑动窗口k遍历该数组，输出得到的最大值数组；

示例1:
输入：nums = [1,3,-1,-3,5,3,6,7], k = 3
输出：[3,3,5,5,6,7]

思路

构造一个单调队列表示当前窗口中单调递减的队列，队列的头就是最大值，为保证这个队列是窗口数据的表示，每次判断队首元素是否是i-k，若是,则pop该队首。

复杂度

时间复杂度:

$O (n)$

空间复杂度:

$O (k)$

Code

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