首页 > 其他 > 详细

每日一题力扣 239 大顶堆

时间:2021-03-28 21:37:18      阅读:23      评论:0      收藏:0      [点我收藏+]

 

给你一个整数数组 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

 

每日一题力扣 239 大顶堆

原文:https://www.cnblogs.com/liuxiangyan/p/14589308.html

(0)
(0)
   
举报
评论 一句话评论(0
关于我们 - 联系我们 - 留言反馈 - 联系我们:wmxa8@hotmail.com
© 2014 bubuko.com 版权所有
打开技术之扣,分享程序人生!