python实现栈和队列
Python利用heapq实现一个优先级队列的方法实现一个优先级队列,每次pop的元素要是优先级高的元素,由于heapq.heapify(list)默认构建一个小顶堆,因此要将priority变为相反数再push,代码如下:
|
import heapq class priorityqueue( object ): """实现一个优先级队列,每次pop优先级最高的元素""" def __init__( self ): self ._queue = [] self ._index = 0 def push( self ,item,priority): heapq.heappush( self ._queue,( - priority, self ._index,item)) #将priority和index结合使用,在priority相同的时候比较index,pop先进入队列的元素 self ._index + = 1 def pop( self ): return heapq.heappop( self ._queue)[ - 1 ] if __name__ = = '__main__' : pqueue = priorityqueue() pqueue.push( 'd' , 4 ) pqueue.push( 'f' , 3 ) pqueue.push( 'a' , 6 ) pqueue.push( 's' , 2 ) print (pqueue.pop()) print (pqueue.pop()) print (pqueue.pop()) |
以上这篇python利用heapq实现一个优先级队列的方法就是小编分享给大家的全部内容了,希望能给大家一个参考,也希望大家多多支持开心学习网。
原文链接:https://blog.csdn.net/Jmiew123/article/details/68951054