Python通过heapq模块实现最小堆,可对列表进行堆化、插入、弹出等操作,支持高效获取极值及模拟最大堆。

Python 中没有专门的“堆”类型,但可以通过内置模块 heapq 来创建和使用堆。heapq 模块提供了对列表进行堆操作的函数,实现的是最小堆(min-heap)结构。
堆在 Python 中本质上是一个普通列表,通过 heapq 提供的函数维护堆的性质。
使用 heapq.heapify() 可将一个无序列表转换为最小堆:
立即学习“Python免费学习笔记(深入)”;
import heapq <p>data = [3, 1, 4, 1, 5, 9, 2, 6] heapq.heapify(data) print(data) # 输出: [1, 1, 2, 3, 5, 9, 4, 6](满足最小堆结构)</p>
使用 heapq.heappush(heap, item) 向堆中添加元素:
立即学习“Python免费学习笔记(深入)”;
heap = [] heapq.heappush(heap, 3) heapq.heappush(heap, 1) heapq.heappush(heap, 4) print(heap) # 输出: [1, 3, 4]
使用 heapq.heappop(heap) 弹出并返回堆顶(最小值):
启科网络商城系统由启科网络技术开发团队完全自主开发,使用国内最流行高效的PHP程序语言,并用小巧的MySql作为数据库服务器,并且使用Smarty引擎来分离网站程序与前端设计代码,让建立的网站可以自由制作个性化的页面。 系统使用标签作为数据调用格式,网站前台开发人员只要简单学习系统标签功能和使用方法,将标签设置在制作的HTML模板中进行对网站数据、内容、信息等的调用,即可建设出美观、个性的网站。
0
立即学习“Python免费学习笔记(深入)”;
min_val = heapq.heappop(heap) print(min_val) # 输出: 1 print(heap) # 输出: [3, 4]
heapq 还提供一些实用函数:
立即学习“Python免费学习笔记(深入)”;
# 示例:获取最小的两个数 nums = [5, 1, 8, 3, 9] two_smallest = heapq.nsmallest(2, nums) print(two_smallest) # 输出: [1, 3]
若需要最大堆,可通过取负值模拟:
max_heap = [] heapq.heappush(max_heap, -10) heapq.heappush(max_heap, -20) largest = -heapq.heappop(max_heap) print(largest) # 输出: 20
基本上就这些。heapq 简单高效,适合实现优先队列、求 TopK、合并多路有序序列等场景。
以上就是python创建和使用堆的方法的详细内容,更多请关注php中文网其它相关文章!
python怎么学习?python怎么入门?python在哪学?python怎么学才快?不用担心,这里为大家提供了python速学教程(入门到精通),有需要的小伙伴保存下载就能学习啦!
Copyright 2014-2025 https://www.php.cn/ All Rights Reserved | php.cn | 湘ICP备2023035733号