首页 > 编程 > Python > 正文

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

2019-11-25 13:19:35
字体:
来源:转载
供稿:网友

实现一个优先级队列,每次pop的元素要是优先级高的元素,由于heapq.heapify(list)默认构建一个小顶堆,因此要将priority变为相反数再push,代码如下:

import heapqclass 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实现一个优先级队列

以上这篇Python利用heapq实现一个优先级队列的方法就是小编分享给大家的全部内容了,希望能给大家一个参考,也希望大家多多支持武林网。

发表评论 共有条评论
用户名: 密码:
验证码: 匿名发表