剑指 Offer 59 - I. 滑动窗口的最大值

Posted 2021-06-21 风去幽墨

题目链接：

https://leetcode-cn.com/problems/hua-dong-chuang-kou-de-zui-da-zhi-lcof/

题意：

给定一个数组 nums 和滑动窗口的大小 k，请找出所有滑动窗口里的最大值

题解：

求最大值，原本想用大顶堆去做的，但是python里好像自己没实现大顶堆只有小顶堆。所以将所有的值取负，转换为求最小值问题。
利用两个小顶堆实现：

• 一个小顶堆用于存储区间内的最小值，一个小顶堆用于存储已不在区间内但还未从前一个小顶堆中删除的值
• 在每次移动窗口时，向堆1中加入新加入的值，向堆2中加入需删除的值
• 在获得当前窗口的最小值的时候，判断两个小顶堆的堆顶元素是否相同，相同则同时删除，直至不同或堆2为空

代码：

class Solution:
    def maxSlidingWindow(self, nums: List[int], k: int) -> List[int]:
        if not nums:return []
        heap,remov,a= [],[],[]
        for i in nums:#取负，转换成找最小值
            a.append(-i)
        for i in range(k):
            heappush(heap,a[i])
        ans = []
        for i in range(k,len(a)+1):
            while len(remov)>0 and remov[0]== heap[0]:
                heappop(remov)
                heappop(heap)
            ans.append(-heap[0])
            if i < len(a):#特殊，最后一个区间
                heappush(remov,a[i-k])
                heappush(heap,a[i])
        return ans