Python利用heapq實(shí)現(xiàn)一個(gè)優(yōu)先級(jí)隊(duì)列的方法

發(fā)布時(shí)間：2020-09-03 09:21:20 來(lái)源：腳本之家閱讀：200 作者：LazyCat_CiCi 欄目：開發(fā)技術(shù)

實(shí)現(xiàn)一個(gè)優(yōu)先級(jí)隊(duì)列，每次pop的元素要是優(yōu)先級(jí)高的元素，由于heapq.heapify(list)默認(rèn)構(gòu)建一個(gè)小頂堆，因此要將priority變?yōu)橄喾磾?shù)再push，代碼如下：

import heapq
class PriorityQueue(object):
  """實(shí)現(xiàn)一個(gè)優(yōu)先級(jí)隊(duì)列，每次pop優(yōu)先級(jí)最高的元素"""
  def __init__(self):
    self._queue = []
    self._index = 0
  def push(self,item,priority):
    heapq.heappush(self._queue,(-priority,self._index,item))#將priority和index結(jié)合使用，在priority相同的時(shí)候比較index，pop先進(jìn)入隊(duì)列的元素
    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實(shí)現(xiàn)一個(gè)優(yōu)先級(jí)隊(duì)列的方法就是小編分享給大家的全部?jī)?nèi)容了，希望能給大家一個(gè)參考，也希望大家多多支持億速云。

向AI問(wèn)一下細(xì)節(jié)

Python利用heapq實(shí)現(xiàn)一個(gè)優(yōu)先級(jí)隊(duì)列的方法

猜你喜歡

最新資訊

相關(guān)推薦

相關(guān)標(biāo)簽