Leetcode-滑动窗口的最大值_python

239.给定一个数组 nums，有一个大小为 k 的滑动窗口从数组的最左侧移动到数组的最右侧。你只可以看到在滑动窗口 k 内的数字。滑动窗口每次只向右移动一位。

返回滑动窗口最大值。

示例:

输入: 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
注意：

你可以假设 k 总是有效的，1 ≤ k ≤ 输入数组的大小，且输入数组不为空。

进阶：

你能在线性时间复杂度内解决此题吗？

思路：

1.根据优先队列的概念，我们假设一个大顶堆，那么一开始的[1,3,-1]，这样一排列成堆的样子就是3在最上面，-1在左下角，1在右下角。。下一步就是[3，-1，-3]了，1就要被挤开了，挤开了也不影响什么，-3再加进来就好了。总之我们需要做的是：(1)维护我们的Heap，也就是删除离开窗口的元素，加入新的元素。这里时间复杂度是logK (2)Max->Top，就是让结果是堆顶的元素。复杂度是O(1)，最后整体的复杂度是NLogK。有没有更好的解法？

扫描二维码关注公众号，回复： 4255151 查看本文章

2.直接用队列，而且是双端队列，也就是两边都能进能出的队列。首先就是入队列，每次滑动窗口都把最大值左边小的数给杀死，也就是出队，后面再滑动窗口进行维护，这样相当于就是每一个数走过场，时间复杂度就是O（N*1），比思路1要小。

class Solution:
    def maxSlidingWindow(self, nums, k):
        """
        :type nums: List[int]
        :type k: int
        :rtype: List[int]
        """
        #严谨判断输入的数字是否合法
        if not nums:return []
        window, res = [], []
        for i, x in enumerate(nums):
            if i>=k and window[0] <= i-k: #窗口滑动时的规律
                window.pop(0)
            while window and nums[window[-1]] <= x: #把最大值左边的数小的就清除。
                window.pop()
            window.append(i)
            if i >= k-1:
                res.append(nums[window[0]])
        return res

Leetcode-滑动窗口的最大值_python

猜你喜欢