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

最新推荐文章于 2024-06-26 19:29:00 发布

LazyCat_CiCi

最新推荐文章于 2024-06-26 19:29:00 发布

阅读量2.4k

点赞数

分类专栏： python 文章标签： python heapq

本文链接：https://blog.csdn.net/Jmiew123/article/details/68951054

版权

本文介绍了如何使用Python的heapq模块创建一个优先级队列，确保每次弹出的元素都是优先级最高的。通过将优先级取反数，实现了在小顶堆中的降序排列。

摘要由CSDN通过智能技术生成

实现一个优先级队列，每次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))