Python实现一个优先级队列的方法


Posted in Python onJuly 31, 2020

问题

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

解决方案

下面的类利用 heapq 模块实现了一个简单的优先级队列:

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]

下面是它的使用方式:

>>> 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)
>>> q.pop()
Item('bar')
>>> q.pop()
Item('spam')
>>> q.pop()
Item('foo')
>>> q.pop()
Item('grok')
>>>

仔细观察可以发现,第一个 pop() 操作返回优先级最高的元素。 另外注意到如果两个有着相同优先级的元素( foogrok ),pop 操作按照它们被插入到队列的顺序返回的。

讨论

这一小节我们主要关注 heapq 模块的使用。 函数 heapq.heappush() heapq.heappop() 分别在队列 _queue 上插入和删除第一个元素, 并且队列 _queue 保证第一个元素拥有最高优先级( 1.4 节已经讨论过这个问题)。 heappop() 函数总是返回”最小的”的元素,这就是保证队列pop操作返回正确元素的关键。 另外,由于 push 和 pop 操作时间复杂度为 O(log N),其中 N 是堆的大小,因此就算是 N 很大的时候它们运行速度也依旧很快。

在上面代码中,队列包含了一个 (-priority, index, item) 的元组。 优先级为负数的目的是使得元素按照优先级从高到低排序。 这个跟普通的按优先级从低到高排序的堆排序恰巧相反。

index 变量的作用是保证同等优先级元素的正确排序。 通过保存一个不断增加的 index 下标变量,可以确保元素按照它们插入的顺序排序。 而且, index 变量也在相同优先级元素比较的时候起到重要作用。

为了阐明这些,先假定 Item 实例是不支持排序的:

>>> a = Item('foo')
>>> b = Item('bar')
>>> a < b
Traceback (most recent call last):
File "<stdin>", line 1, in <module>
TypeError: unorderable types: Item() < Item()
>>>

如果你使用元组 (priority, item) ,只要两个元素的优先级不同就能比较。 但是如果两个元素优先级一样的话,那么比较操作就会跟之前一样出错:

>>> a = (1, Item('foo'))
>>> b = (5, Item('bar'))
>>> a < b
True
>>> c = (1, Item('grok'))
>>> a < c
Traceback (most recent call last):
File "<stdin>", line 1, in <module>
TypeError: unorderable types: Item() < Item()
>>>

通过引入另外的 index 变量组成三元组 (priority, index, item) ,就能很好的避免上面的错误, 因为不可能有两个元素有相同的 index 值。Python 在做元组比较时候,如果前面的比较已经可以确定结果了, 后面的比较操作就不会发生了:

>>> a = (1, 0, Item('foo'))
>>> b = (5, 1, Item('bar'))
>>> c = (1, 2, Item('grok'))
>>> a < b
True
>>> a < c
True
>>>

如果你想在多个线程中使用同一个队列,那么你需要增加适当的锁和信号量机制。 可以查看 12.3 小节的例子演示是怎样做的。

heapq 模块的官方文档有更详细的例子程序以及对于堆理论及其实现的详细说明。

以上就是Python实现一个优先级队列的方法的详细内容,更多关于Python实现优先级队列的资料请关注三水点靠木其它相关文章!

Python 相关文章推荐
python3 与python2 异常处理的区别与联系
Jun 19 Python
Python+Selenium+PIL+Tesseract自动识别验证码进行一键登录
Sep 20 Python
Python给图像添加噪声具体操作
Mar 03 Python
Python爬取豆瓣视频信息代码实例
Nov 16 Python
用python的turtle模块实现给女票画个小心心
Nov 23 Python
Python基础之字符串操作常用函数集合
Feb 09 Python
Pycharm+Python工程,引用子模块的实现
Mar 09 Python
python matplotlib.pyplot.plot()参数用法
Apr 14 Python
浅谈tensorflow中dataset.shuffle和dataset.batch dataset.repeat注意点
Jun 08 Python
python自动生成sql语句的脚本
Feb 24 Python
python编程实现清理微信重复缓存文件
Nov 01 Python
Python下载商品数据并连接数据库且保存数据
Mar 31 Python
django表单中的按钮获取数据的实例分析
Jul 31 #Python
pycharm中使用request和Pytest进行接口测试的方法
Jul 31 #Python
django创建css文件夹的具体方法
Jul 31 #Python
Selenium之模拟登录铁路12306的示例代码
Jul 31 #Python
python的flask框架难学吗
Jul 31 #Python
使用PyCharm安装pytest及requests的问题
Jul 31 #Python
django和flask哪个值得研究学习
Jul 31 #Python
You might like
PHP 数组教程 定义数组
2009/10/23 PHP
php实现猴子选大王问题算法实例
2015/04/20 PHP
php中define用法实例
2015/07/30 PHP
php 替换文章中的图片路径,下载图片到本地服务器的方法
2018/02/06 PHP
event.currentTarget与event.target的区别介绍
2012/12/31 Javascript
ie6下png图片背景不透明的解决办法使用js实现
2013/01/11 Javascript
js实现一个可以兼容PC端和移动端的div拖动效果实例
2016/12/09 Javascript
详解自动生成博客目录案例
2016/12/09 Javascript
jQuery编写网页版2048小游戏
2017/01/06 Javascript
Javascript 详解封装from表单数据为json串进行ajax提交
2017/03/29 Javascript
深入理解Angular中的依赖注入
2017/06/26 Javascript
vue组件Prop传递数据的实现示例
2017/08/17 Javascript
three.js实现3D模型展示的示例代码
2017/12/31 Javascript
微信小程序滑动选择器的实现代码
2018/08/10 Javascript
小程序实现单选多选功能
2018/11/04 Javascript
JS实现可用滑块滑动的缓动图代码
2019/09/01 Javascript
[01:06:25]Secret vs Liquid 2018国际邀请赛淘汰赛BO3 第一场 8.25
2018/08/29 DOTA
python range()函数取反序遍历sequence的方法
2018/06/25 Python
python找出完数的方法
2018/11/12 Python
从numpy数组中取出满足条件的元素示例
2019/11/26 Python
python3的UnicodeDecodeError解决方法
2019/12/20 Python
python使用配置文件过程详解
2019/12/28 Python
linux环境下安装python虚拟环境及注意事项
2020/01/07 Python
python代码区分大小写吗
2020/06/17 Python
Django如何使用asyncio协程和ThreadPoolExecutor多线程
2020/10/12 Python
html5中canvas学习笔记2-判断浏览器是否支持canvas
2013/01/06 HTML / CSS
杭州SQL浙江浙大网新恩普软件有限公司
2013/07/27 面试题
工商管理专业实习生自我鉴定
2013/09/29 职场文书
材料采购员岗位职责
2013/12/17 职场文书
中学生自我鉴定
2014/02/04 职场文书
中华美德颂演讲稿
2014/05/20 职场文书
咖啡厅商业计划书
2014/09/15 职场文书
个人租房协议书范本
2014/09/30 职场文书
工作自我评价范文
2015/03/05 职场文书
红歌会主持词
2015/07/02 职场文书
Redis过期数据是否会被立马删除
2022/07/23 Redis