[Python]heapq 堆模块

模块的名称是 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怎么更新,总可以取出最值。

你可能感兴趣的:(算法结构,如今Python,python,堆,heapd)