[Python]heapq 堆模块

最新推荐文章于 2024-06-26 19:29:00 发布

orangleliu

最新推荐文章于 2024-06-26 19:29:00 发布

阅读量6.2k

点赞数

分类专栏：算法结构如今Python 文章标签： python 堆 heapd

本文为orangleliu (http://blog.csdn.net/orangleliu/)原创文章，自由传播, 文章转载请声明, 多谢。

本文链接：https://blog.csdn.net/orangleliu/article/details/24923539

版权

如今Python 同时被 2 个专栏收录

174 篇文章 4 订阅

订阅专栏

算法结构

8 篇文章 0 订阅

订阅专栏

模块的名称是 heapq

如果一个比较大的列表，而且是要不断的更新数据，而你需要不停的从其中取出最大或者是最小的值。一种简单的方法就是每次，更新list之后就对其进行sort操作，但是如果list很大，这样每次排序不仅效率低下，资源消耗也相当的大。

使用场景：优先级队列

这里就是对一个list 不断更新，得到最值

>>> import heapq
>>> a = []
>>> heapq.heappush(a, 4)
>>> heapq.heappush(a, 5)
>>> heapq.heappush(a, 1)
>>> heapq.heappush(a, 100)
>>> heapq.heappush(a, 33)
>>> heapq.heappop(a)
1
>>> heapq.heappush(a, 3)
>>> heapq.heappop(a)
3
>>> heapq.nlargest(1, a)
[100]
>>> heapq.nsmallest(1, a)
[4]
>>> heapq.heappush(a, 5)
>>> heapq.nsmallest(2, a)
[4, 5]

不管list怎么更新，总可以取出最值。