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内置函数Type()函数一个有趣的用法
Feb 18 Python
python通过smpt发送邮件的方法
Apr 30 Python
Python和Perl绘制中国北京跑步地图的方法
Mar 03 Python
python生成随机图形验证码详解
Nov 08 Python
Python中的Django基本命令实例详解
Jul 15 Python
Python3.5 处理文本txt,删除不需要的行方法
Dec 10 Python
如何用python写一个简单的词法分析器
Dec 18 Python
django之对FileField字段的upload_to的设定方法
Jul 28 Python
Python内置加密模块用法解析
Nov 25 Python
浅谈tensorflow模型保存为pb的各种姿势
May 25 Python
一篇文章带你搞定Ubuntu中打开Pycharm总是卡顿崩溃
Nov 02 Python
python 实现网易邮箱邮件阅读和删除的辅助小脚本
Mar 01 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
set_include_path在win和linux下的区别
2008/01/10 PHP
基于empty函数的输出详解
2013/06/17 PHP
php中常见的sql攻击正则表达式汇总
2014/11/06 PHP
20个非常有用的PHP类库 加速php开发
2010/01/15 Javascript
jQuery 在光标定位的地方插入文字的插件
2012/05/10 Javascript
原生JavaScript实现连连看游戏(附源码)
2013/11/05 Javascript
取得元素的左和上偏移量的方法
2014/09/17 Javascript
基于jquery二维码生成插件qrcode
2017/01/07 Javascript
Angular.js中angular-ui-router的简单实践
2017/07/18 Javascript
深入理解Vue 组件之间传值
2018/08/16 Javascript
深入解析Vue源码实例挂载与编译流程实现思路详解
2019/05/05 Javascript
微信小程序在text文本实现多种字体样式
2019/11/08 Javascript
微信小程序获取复选框全选反选选中的值(实例代码)
2019/12/17 Javascript
vue中的使用token的方法示例
2020/03/10 Javascript
python实现简单socket程序在两台电脑之间传输消息的方法
2015/03/13 Python
如何使用七牛Python SDK写一个同步脚本及使用教程
2015/08/23 Python
Python图算法实例分析
2016/08/13 Python
Python Socket编程详细介绍
2017/03/23 Python
python环形单链表的约瑟夫问题详解
2018/09/27 Python
深入浅析Python 命令行模块 Click
2020/03/11 Python
python给视频添加背景音乐并改变音量的具体方法
2020/07/19 Python
Python jieba库分词模式实例用法
2021/01/13 Python
在Pycharm中安装Pandas库方法(简单易懂)
2021/02/20 Python
英国办公用品商店:Office Outlet
2018/04/04 全球购物
莱德杯高尔夫欧洲官方商店:Ryder Cup Shop
2019/08/14 全球购物
介绍一下grep命令的使用
2012/06/28 面试题
如果让你测试一台高速激光打印机,你都会进行哪些测试
2012/12/04 面试题
装饰活动策划方案
2014/02/11 职场文书
便利店促销方案
2014/02/20 职场文书
劲霸男装广告词改编版
2014/03/21 职场文书
关于感恩的演讲稿400字
2014/08/26 职场文书
委托书的样本
2015/01/28 职场文书
小班上学期个人总结
2015/02/12 职场文书
《学会生存》读后感3篇
2019/12/09 职场文书
Django中的JWT身份验证的实现
2021/05/07 Python
详解CSS不受控制的position fixed
2021/05/25 HTML / CSS