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的Django框架中创建语言文件
Jul 27 Python
Python实现连接postgresql数据库的方法分析
Dec 27 Python
Python实现输入二叉树的先序和中序遍历,再输出后序遍历操作示例
Jul 27 Python
Python爬虫学习之获取指定网页源码
Jul 30 Python
Python从入门到精通之环境搭建教程图解
Sep 26 Python
Python 解决相对路径问题:&quot;No such file or directory&quot;
Jun 05 Python
keras打印loss对权重的导数方式
Jun 10 Python
Selenium自动化测试工具使用方法汇总
Jun 12 Python
python元组拆包实现方法
Feb 28 Python
浅谈Python列表嵌套字典转化的问题
Apr 07 Python
python程序的组织结构详解
Dec 06 Python
Python 中面向接口编程
May 20 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 出现乱码和Sessions验证问题的解决方法!
2008/12/06 PHP
深入php之规范编程命名小结
2013/05/15 PHP
php跨域cookie共享使用方法
2014/02/20 PHP
跟我学Laravel之快速入门
2014/10/15 PHP
PHP file_get_contents函数读取远程数据超时的解决方法
2015/05/13 PHP
用PHP写的一个冒泡排序法的函数简单实例
2016/05/26 PHP
php自定义函数实现二维数组按指定key排序的方法
2016/09/29 PHP
PHP实现超简单的SSL加密解密、验证及签名的方法示例
2017/08/28 PHP
php 使用ActiveMQ发送消息,与处理消息操作示例
2020/02/23 PHP
javascript new后的constructor属性
2010/08/05 Javascript
基于jquery实现页面滚动时顶部导航显示隐藏
2020/04/20 Javascript
js滚动条平滑移动示例代码
2016/03/29 Javascript
jquery实现左右无缝轮播图
2020/07/31 Javascript
Ext JS动态加载JavaScript创建窗体的方法
2016/06/23 Javascript
jQuery内存泄露解决办法
2016/12/13 Javascript
详解实现一个通用的“划词高亮”在线笔记功能
2019/04/23 Javascript
关于NodeJS中的循环引用详解
2019/07/23 NodeJs
微信小程序列表时间戳转换实现过程解析
2019/10/12 Javascript
Vue 组件的挂载与父子组件的传值实例
2020/09/02 Javascript
Python序列之list和tuple常用方法以及注意事项
2015/01/09 Python
python获取图片颜色信息的方法
2015/03/18 Python
python下如何查询CS反恐精英的服务器信息
2017/01/17 Python
使用Python进行AES加密和解密的示例代码
2018/02/02 Python
VSCode下好用的Python插件及配置
2018/04/06 Python
python3实现小球转动抽奖小游戏
2020/04/15 Python
在pycharm中为项目导入anacodna环境的操作方法
2020/02/12 Python
肯尼亚网上商城:Kilimall
2016/08/20 全球购物
澳洲的服装老品牌:SABA
2018/02/06 全球购物
物业经理求职自我评价
2013/09/22 职场文书
祖国在我心中演讲稿600字
2014/05/04 职场文书
大班亲子运动会方案
2014/06/10 职场文书
爱祖国爱家乡演讲稿
2014/09/02 职场文书
学校领导四风问题整改措施思想汇报
2014/10/09 职场文书
学生党员批评与自我批评
2014/10/15 职场文书
感谢师恩主题班会
2015/08/17 职场文书
中国梦党课学习心得体会
2016/01/05 职场文书