优先队列(Python)

优先队列与队列一样,只能从队尾插入元素,从队首删除元素。但是它有一个特性,就是队列中最大或者最小的元素总是位于队首,所以出队时,并非按照先进先出的原则进行,而是将当前队列中最大或者最小的元素出队。这点类似于给队列里的元素进行了排序。
我在这使用Python内置的一个库使用一下优先队列,上次提到的Dijkstra求最短路径的算法,就用到了优先队列。

import heapq

pqueue = []
heapq.heappush(pqueue,(0,"A"))
heapq.heappush(pqueue,(2,"B"))
heapq.heappush(pqueue,(10,"C"))
heapq.heappush(pqueue,(6,"D"))
heapq.heappush(pqueue,(15,"E"))
print(pqueue)
"""
result:
	[(0, 'A'), (2, 'B'), (10, 'C'), (6, 'D'), (15, 'E')]
"""

上面是我给优先队列里面添加元素,下面我们从优先队列里面弹出元素,请看

print(heapq.heappop(pqueue))
print(heapq.heappop(pqueue))
print(heapq.heappop(pqueue))
print(heapq.heappop(pqueue))
print(heapq.heappop(pqueue))
"""
results:
	(0, 'A')
	(2, 'B')
	(6, 'D')
	(10, 'C')
	(15, 'E')
"""

这里弹出的顺序,就是按照优先队列里面元素对应权值的大小弹出的。

你可能感兴趣的:(数据结构与算法,python,队列)