Python cookbook(數(shù)據(jù)結(jié)構(gòu)與算法)實(shí)現(xiàn)優(yōu)先級(jí)隊(duì)列的方法示例
本文實(shí)例講述了Python實(shí)現(xiàn)優(yōu)先級(jí)隊(duì)列的方法。分享給大家供大家參考,具體如下:
問(wèn)題:要實(shí)現(xiàn)一個(gè)隊(duì)列,它能夠以給定的優(yōu)先級(jí)對(duì)元素排序,且每次pop操作時(shí)都會(huì)返回優(yōu)先級(jí)最高的那個(gè)元素;
解決方案:采用heapq模塊實(shí)現(xiàn)一個(gè)簡(jiǎn)單的優(yōu)先級(jí)隊(duì)列
# example.py # # Example of a priority queue import heapq class PriorityQueue: def __init__(self): self._queue = [] self._index = 0 def push(self, item, priority): heapq.heappush(self._queue, (-priority, self._index, item)) self._index += 1 def pop(self): return heapq.heappop(self._queue)[-1] # Example use 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) print("Should be bar:", q.pop()) print("Should be spam:", q.pop()) print("Should be foo:", q.pop()) print("Should be grok:", q.pop())
Python 3.4.0 (v3.4.0:04f714765c13, Mar 16 2014, 19:24:06) [MSC v.1600 32 bit (Intel)] on win32 Type "copyright", "credits" or "license()" for more information. >>> ================================ RESTART ================================ >>> Should be bar: Item('bar') Should be spam: Item('spam') Should be foo: Item('foo') Should be grok: Item('grok') >>>
可以看出:第一次執(zhí)行pop()操作時(shí)返回的元素具有最高的優(yōu)先級(jí);對(duì)于相同優(yōu)先級(jí)的兩個(gè)元素(foo和gork)返回的順序同它們插入到隊(duì)列時(shí)的順序相同。
在這段代碼中,隊(duì)列以元組(-priority, self._index, item)的形式組成,priority取負(fù)值是為了隊(duì)列按照從高到低的順序排列,這和堆默認(rèn)的從小到大的排序相反。
變量index的作用是對(duì)相同優(yōu)先級(jí)的元素以適當(dāng)?shù)捻樞蚺帕?,特別對(duì)同優(yōu)先級(jí)的元素間做比較操作時(shí)扮演了重要的角色。
Item實(shí)例無(wú)法進(jìn)行次序比較:
a=Item('foo') b=Item('bar') print('a<b: ',a<b) >>> Traceback (most recent call last): File "D:\4autotests\02script\python-cookbook\python-cookbook-master\src\1\5.implementing_a_priority_queue\example.py", line 27, in <module> print('a<b: ',a<b) TypeError: unorderable types: Item() < Item() >>>
如果以元組(priority, item)的形式來(lái)表示元素,只要優(yōu)先級(jí)不同,就可進(jìn)行比較:
a=(1,Item('foo')) b=(5,Item('bar')) c=(1,Item('gork')) print('a<b: ',a<b) print('a<c: ',a<c) >>> a<b: True Traceback (most recent call last): File "D:\4autotests\02script\python-cookbook\python-cookbook-master\src\1\5.implementing_a_priority_queue\example.py", line 29, in <module> print('a<c: ',a<c) TypeError: unorderable types: Item() < Item() >>>
引入額外的索引值,以(priority, index, item)的方式建立元組,就可以避免相同優(yōu)先級(jí)無(wú)法比較的問(wèn)題,因?yàn)闆](méi)有哪兩個(gè)元組會(huì)有相同的index值;
a=(1,0,Item('foo')) b=(5,1,Item('bar')) c=(1,2,Item('gork')) print('a<b: ',a<b) print('a<c: ',a<c) >>> a<b: True a<c: True >>>
如果想將這個(gè)隊(duì)列用于線程間通信,還需要增加適當(dāng)?shù)逆i和信號(hào)機(jī)制。
(代碼摘自《Python Cookbook》)
更多關(guān)于Python相關(guān)內(nèi)容感興趣的讀者可查看本站專題:《Python數(shù)據(jù)結(jié)構(gòu)與算法教程》、《Python加密解密算法與技巧總結(jié)》、《Python編碼操作技巧總結(jié)》、《Python函數(shù)使用技巧總結(jié)》、《Python字符串操作技巧匯總》及《Python入門(mén)與進(jìn)階經(jīng)典教程》
希望本文所述對(duì)大家Python程序設(shè)計(jì)有所幫助。
相關(guān)文章
Python中Celery異步任務(wù)隊(duì)列的具體使用
Celery是一個(gè)用于處理分布式任務(wù)和作業(yè)隊(duì)列的異步任務(wù)隊(duì)列庫(kù),本文主要介紹了Python中Celery異步任務(wù)隊(duì)列的具體使用,文中通過(guò)示例代碼介紹的非常詳細(xì),需要的朋友們下面隨著小編來(lái)一起學(xué)習(xí)學(xué)習(xí)吧2024-02-02Django+zTree構(gòu)建組織架構(gòu)樹(shù)的方法
zTree是一個(gè)開(kāi)源的依靠JQuery實(shí)現(xiàn)的多功能樹(shù)插件,具有性能優(yōu)異、配置靈活、功能強(qiáng)大的特點(diǎn),這篇文章主要介紹了Django+zTree構(gòu)建組織架構(gòu)樹(shù),需要的朋友可以參考下2019-08-08python提取照片坐標(biāo)信息的實(shí)例代碼
這篇文章主要介紹了python提取照片坐標(biāo)信息的實(shí)例代碼,文中給大家提到了Python利用exifread庫(kù)來(lái)解析照片的經(jīng)緯度,通過(guò)實(shí)例代碼給大家介紹的非常詳細(xì),具有一定的參考借鑒價(jià)值,需要的朋友可以參考下2019-08-08對(duì)Django中的權(quán)限和分組管理實(shí)例講解
今天小編就為大家分享一篇對(duì)Django中的權(quán)限和分組管理實(shí)例講解,具有很好的參考價(jià)值,希望對(duì)大家有所幫助。一起跟隨小編過(guò)來(lái)看看吧2019-08-08Python實(shí)現(xiàn)的Google IP 可用性檢測(cè)腳本
這篇文章主要介紹了Python實(shí)現(xiàn)的Google IP 可用性檢測(cè)腳本,本文腳本需要Python 3.4+環(huán)境,需要的朋友可以參考下2015-04-04Python中selenium實(shí)現(xiàn)文件上傳所有方法整理總結(jié)
本篇文章主要介紹了Python中selenium實(shí)現(xiàn)文件上傳所有方法整理總結(jié),具有一定的參考價(jià)值,感興趣的小伙伴們可以參考一下。2017-04-04Python隨機(jī)生成一個(gè)6位的驗(yàn)證碼代碼分享
這篇文章主要介紹了Python隨機(jī)生成一個(gè)6位的驗(yàn)證碼代碼分享,本文直接給出代碼實(shí)例,需要的朋友可以參考下2015-03-03Python3.6實(shí)現(xiàn)根據(jù)電影名稱(支持電視劇名稱),獲取下載鏈接的方法
這篇文章主要介紹了Python3.6實(shí)現(xiàn)根據(jù)電影名稱(支持電視劇名稱),獲取下載鏈接的方法,涉及Python爬蟲(chóng)與正則相關(guān)操作技巧,需要的朋友可以參考下2019-08-08