用 python 实现大小堆 & Top K 问题

用 python 实现大小堆 & Top K 问题

python的 heapq 模块提供了堆队列(优先队列)的实现,默认是最小堆,即父节点不小于任意一个子节点
使用最小堆可以实现前 K 大的问题,原理是维护一个长度为 K 的堆,每插入一个新元素就弹出之前的最小值,从而最终只剩最大的 K 个值
如想解决前 K 小的问题,则需要将所有元素取负后加入堆,这样最终维持的是所有元素负数中最大的 K 个,再次取负后则得到最小的 K 个元素

案例题目:
1509. Minimum Difference Between Largest and Smallest Value in Three Moves
Top K 问题,找到最大的4个数和最小的4个数,删掉

  • 最大的三个
  • 最小的三个
  • 最大的两个和最小的一个
  • 最大的一个和最小的两个

然后用剩下的最大数减最小数,取四种情况中最小的结果返回

def minDifference(self, nums: List[int]) -> int:
    
    if len(nums) <= 4:
        return 0
    
    minheap = []
    maxheap = []
    
    for num in nums:
        
        if len(minheap
  • 0
    点赞
  • 2
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值