用 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
    评论
Python中,topk是一种常见的算法,用于从一个列表或数组中找出前k个最大或最小的元素。引用\[1\]中的代码展示了一个对province_room_quality_data对象进行TopK排序的算法。该算法使用了排序的思想,通过维护一个大小为k的最小堆实现。首先,将前k个元素构建成一个最小堆。然后,遍历剩余的元素,如果当前元素的quality大于顶元素的quality,则将顶元素弹出,将当前元素插入中。最后,将中的元素转换为列表并返回。这样就得到了前k个quality最大的province_room_quality_data对象。引用\[2\]中的代码展示了另一种选择算法,用于从一个序列中选择第k小的元素。该算法使用了快速排序的思想,通过递归地将序列划分为小于主元的部分和大于主元的部分,直到找到第k小的元素。引用\[3\]中的代码展示了使用排序和冒泡排序两种算法对一个包含100000个元素的列表进行topk操作的执行时间对比。结果显示,排序算法的执行时间为0.005194902420043945秒,而冒泡排序算法的执行时间为0.9621851444244385秒。因此,可以看出排序算法在处理大规模数据时具有更高的效率。 #### 引用[.reference_title] - *1* [python 实现 topk算法](https://blog.csdn.net/qq_42604176/article/details/122653275)[target="_blank" data-report-click={"spm":"1018.2226.3001.9630","extra":{"utm_source":"vip_chatgpt_common_search_pc_result","utm_medium":"distribute.pc_search_result.none-task-cask-2~all~insert_cask~default-1-null.142^v91^insertT0,239^v3^insert_chatgpt"}} ] [.reference_item] - *2* [python TopK算法](https://blog.csdn.net/xingtianyao/article/details/101053321)[target="_blank" data-report-click={"spm":"1018.2226.3001.9630","extra":{"utm_source":"vip_chatgpt_common_search_pc_result","utm_medium":"distribute.pc_search_result.none-task-cask-2~all~insert_cask~default-1-null.142^v91^insertT0,239^v3^insert_chatgpt"}} ] [.reference_item] - *3* [python-排序算法(四)、排序之topk问题](https://blog.csdn.net/adminwg/article/details/126816340)[target="_blank" data-report-click={"spm":"1018.2226.3001.9630","extra":{"utm_source":"vip_chatgpt_common_search_pc_result","utm_medium":"distribute.pc_search_result.none-task-cask-2~all~insert_cask~default-1-null.142^v91^insertT0,239^v3^insert_chatgpt"}} ] [.reference_item] [ .reference_list ]

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

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

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值