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 相关文章推荐
对Python新手编程过程中如何规避一些常见问题的建议
Apr 01 Python
Python运算符重载用法实例分析
Jun 01 Python
Python环境搭建之OpenCV的步骤方法
Oct 20 Python
Python装饰器(decorator)定义与用法详解
Feb 09 Python
python绘制简单彩虹图
Nov 19 Python
python实现事件驱动
Nov 21 Python
Python3.0中普通方法、类方法和静态方法的比较
May 03 Python
PyQt5 控件字体样式等设置的实现
May 13 Python
Python新手如何进行闭包时绑定变量操作
May 29 Python
Python OpenCV读取中文路径图像的方法
Jul 02 Python
详解Python中import机制
Sep 11 Python
Pytorch中Softmax与LogSigmoid的对比分析
Jun 05 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
在WIN98下以apache模块方式安装php
2006/10/09 PHP
解析mysql 表中的碎片产生原因以及清理
2013/06/22 PHP
php获取字段名示例分享
2014/03/03 PHP
关于php微信订阅号开发之token验证后自动发送消息给订阅号但是没有消息返回的问题
2015/12/21 PHP
Yii2中如何使用modal弹窗(基本使用)
2016/05/30 PHP
PHP入门教程之操作符与控制结构流程详解
2016/09/09 PHP
简单实现php上传文件功能
2017/09/21 PHP
js substr、substring和slice使用说明小记
2011/09/15 Javascript
用JQuery 判断某个属性是否存在hasAttr的解决方法
2013/04/26 Javascript
js获取上传文件大小示例代码
2014/04/10 Javascript
jQuery的:parent选择器定义和用法
2014/07/01 Javascript
JavaScript识别网页关键字并进行描红的方法
2015/11/09 Javascript
Jquery Ajax Error 调试错误的技巧
2015/11/20 Javascript
Angular之指令Directive用法详解
2017/03/01 Javascript
将input框中输入内容显示在相应的div中【三种方法可选】
2017/05/08 Javascript
jQuery dateRangePicker插件使用方法详解
2017/07/28 jQuery
Express使用html模板的详细代码
2017/09/18 Javascript
微信小程序使用navigateTo数据传递的实例
2017/09/26 Javascript
Node.js模拟发起http请求从异步转同步的5种用法
2018/09/26 Javascript
vuex如何重置所有state(可定制)
2019/01/17 Javascript
详解ng-alain动态表单SF表单项设置必填和正则校验
2019/06/11 Javascript
vue实现导航菜单和编辑文本的示例代码
2020/07/04 Javascript
vue 解决无法对未定义的值,空值或基元值设置反应属性报错问题
2020/07/31 Javascript
python实现求解列表中元素的排列和组合问题
2018/03/15 Python
python获取命令行输入参数列表的实例代码
2018/06/23 Python
Python3.6实现根据电影名称(支持电视剧名称),获取下载链接的方法
2019/08/26 Python
css3圆角边框和边框阴影示例
2014/05/05 HTML / CSS
艺龙旅行网酒店预订:国内、港澳台酒店
2018/06/26 全球购物
文明宿舍获奖感言
2014/02/07 职场文书
先进单位事迹材料
2014/12/25 职场文书
旷工检讨书1000字
2015/01/01 职场文书
2016年暑假学生家长评语
2015/12/01 职场文书
2016高中社会实践心得体会范文
2016/01/14 职场文书
导游词之镇江焦山
2019/11/21 职场文书
晶体管单管来复再生式收音机
2021/04/22 无线电
nginx 配置缓存
2022/05/11 Servers