【Leetcode堆和双端队列】滑动窗口最大值(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 ≤ 输入数组的大小。

进阶:
你能在线性时间复杂度内解决此题吗?

解答

方法一:暴力破解。cur表示窗口,记录最大值,依次后移。时间复杂度O(n·k),太慢了,才超过30%的代码

方法二:用大顶堆存窗口元素,堆顶就是当前窗口最大值,依次后移,删除最左边,加入一个新元素,堆删除操作时间复杂度O(log(n)),插入操作O(1),这种方法总的时间复杂度O(n·log(k))。
But!Python中没有大顶堆的实现,用“存入堆、从堆中取出的时候,都用相反数”模拟大顶堆可行,以为这就解决了吗? No! heapq包可以删除堆顶元素,但没提供删除堆中某元素的方法,所有先找到元素在堆中位置再用了del删除,del删除后,存储堆的列表顺序就不是堆了,需要重新排列成堆,时间复杂度O(n),所以用Python最后做出来时间复杂度O(n·k),但这个提交速度为毛没有暴力破解快,无语。。。

方法三:双向队列,时间复杂度O(n)
思路:依次遍历列表中的元素,cur存放已经在窗口中的元素的下标,假如当前元素为c, 则用c和cur中元素从后向前比较,删除cur中小于c的元素,并把c的下标加到cur中,这步操作保证了cur[0]表示的元素永远是窗口中最大的
注:c和cur比较时,一定要从后往前,否则部分样例不能通过:[1, 3, 1, 2, 0, 5], k=3 .........................视频里讲的从前往后:)

通过代码如下:

from heapq import *


class Solution:

    # 方法一:暴力破解
    # def maxSlidingWindow(self, nums: List[int], k: int) -> List[int]:
    #     if not nums:
    #         return []
    #     ans = []
    #     cur = []
    #     for x in range(k):
    #         cur.append(nums[x])

    #     ans.append(max(cur))
    #     while x < len(nums)-1:
    #         del cur[0]              # 删除表头
    #         cur.append(nums[x+1])   # 添加到尾
    #         ans.append(max(cur))    # max()的平均时间复杂度是O(n)
    #         x = x+1
    #     return ans

    # 方法二:大顶堆
    # def maxSlidingWindow(self, nums: List[int], k: int) -> List[int]:
    #     if not nums:
    #         return []
    #     l = []                                  # 预存堆
    #     ans = []
    #     for index, x in enumerate(nums):
    #         if index==0 or len(l)<k:            # 堆还没满,入堆
    #             heappush(l, -x)
    #         else:
    #             ans.append(-l[0])               # 堆顶,最小的,取负数刚好是最大的
    #             del l[l.index(-nums[index-k])]  # 删除窗口左侧第一个
    #             heappush(l, -x)                 # 入堆
    #             heapify(l)                      # 堆貌似乱了......, 重新排列成堆,时间复杂度是O(n)。。。
    #     ans.append(-l[0])
    #     return ans

    # 方法三:双端队列
    def maxSlidingWindow(self, nums, k):
        if not nums:
            return []
        cur = []                                    # 存窗口元素的index,cur[0]位置永远是最大的
        ans = []
        for index, x in enumerate(nums):
            if index >= k and index - k >= cur[0]:  # cur[0]和当前元素的index距离为k,说明cur[0]该从窗口删除了
                cur.pop(0)
            while cur and x >= nums[cur[-1]]:       # 这里,cur要从后往前删小于当前元素的值
                cur.pop()
            cur.append(index)

            if index + 1 >= k:                      # 当前元素位已经走过了窗口距离大小
                ans.append(nums[cur[0]])
        return ans

原文地址:https://www.cnblogs.com/ldy-miss/p/11991143.html