您的位置:首页 > 编程语言 > Python开发

Python利用heapq实现一个优先级队列

2017-04-02 10:43 761 查看
实现一个优先级队列,每次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