您好,登錄后才能下訂單哦!
實現一個優先級隊列,每次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實現一個優先級隊列的方法就是小編分享給大家的全部內容了,希望能給大家一個參考,也希望大家多多支持億速云。
免責聲明:本站發布的內容(圖片、視頻和文字)以原創、轉載和分享為主,文章觀點不代表本網站立場,如果涉及侵權請聯系站長郵箱:is@yisu.com進行舉報,并提供相關證據,一經查實,將立刻刪除涉嫌侵權內容。