每日一题力扣 239 大顶堆

 

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

返回滑动窗口中的最大值。

来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/sliding-window-maximum
著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。

 

class Solution:
    def maxSlidingWindow(self, nums: List[int], k: int) -> List[int]:
        n=len(nums)
        q=[(-nums[i],i)for i in range(k)]#找到前k个并且取负数，记录下标值
        heapq.heapify(q)#将列表建成小顶堆
        ans=[-q[0][0]]#记录顶值，取负回来
        for i in range(k,n):
            heapq.heappush(q,(-nums[i],i))#将下一个拿入
            while q[0][1]<=i-k:
                heapq.heappop(q)#剔除顶
            ans.append(-q[0][0])#否则就把它给加入
        return ans