1.5 实现一个优先级队列

最新推荐文章于 2023-04-02 16:18:13 发布

yepeng0914

最新推荐文章于 2023-04-02 16:18:13 发布

阅读量121

点赞数

分类专栏： Python基础

本文链接：https://blog.csdn.net/Tangerine02/article/details/102996144

版权

Python基础专栏收录该内容

14 篇文章 0 订阅

订阅专栏

问题
怎样实现一个按优先级排序的队列？并且在这个队列上面每次pop操作总是返回优先级最
高的那个元素

"""
通过引入另外的 index 变量组成三元组 (priority, index, item) ，
就能很好的避免上面 的错误， 因为不可能有两个元素有相同的 index 值。
"""

import  heapq

class PriorityQueue:
    def __init__(self):
        self._queque=[]
        self._index=0

    def push(self,item,priority):
        heapq.heappush(self._queque,(-priority,self._index,item))
        self._index+=1
    def pop(self):
        return heapq.heappop(self._queque)[-1]

class Item:
     def __init__(self,name):
         self.name=name
     def __repr__(self):
         return  "Item({!r})".format(self.name)


q = PriorityQueue()
q.push(Item('foo'), 1)
q.push(Item('bar'), 5)
q.push(Item('spam'), 4)
q.push(Item('grok'), 1)