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中if __name__ == &quot;__main__&quot;详细解释
Oct 21 Python
编写Python脚本使得web页面上的代码高亮显示
Apr 24 Python
python正则实现计算器功能
Dec 14 Python
numpy自动生成数组详解
Dec 15 Python
对Python中for复合语句的使用示例讲解
Nov 01 Python
flask session组件的使用示例
Dec 25 Python
PyQt4实时显示文本内容GUI的示例
Jun 14 Python
PyQt5 QTable插入图片并动态更新的实例
Jun 18 Python
对Django 中request.get和request.post的区别详解
Aug 12 Python
tensorflow保持每次训练结果一致的简单实现
Feb 17 Python
将数据集制作成VOC数据集格式的实例
Feb 17 Python
Python必备技巧之字符数据操作详解
Mar 23 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
利用谷歌 Translate API制作自己的翻译脚本
2014/06/04 PHP
Laravel 5 框架入门(一)
2015/04/09 PHP
jQuery 页面 Mask实现代码
2010/01/09 Javascript
jquery中对表单的基本操作代码
2010/07/29 Javascript
JavaScript 模式之工厂模式(Factory)应用介绍
2012/11/15 Javascript
Function.prototype.call.apply结合用法分析示例
2013/07/03 Javascript
JQuery判断HTML元素是否存在的两种解决方法
2013/12/26 Javascript
jQuery中bind与live的用法及区别小结
2014/01/27 Javascript
谈谈我对JavaScript DOM事件的理解
2015/12/18 Javascript
JavaScript 不支持 indexof 该如何解决
2016/03/30 Javascript
基于JSON格式数据的简单jQuery幻灯片插件(jquery-slider)
2016/08/10 Javascript
js实现目录链接,内容跟着目录滚动显示的简单实例
2016/10/15 Javascript
nodejs学习笔记之路由
2017/03/27 NodeJs
jQuery+pjax简单示例汇总
2017/04/21 jQuery
JS设置手机验证码60s等待实现代码
2017/06/14 Javascript
简单实现js拖拽效果
2017/07/25 Javascript
Nuxt.js踩坑总结分享
2018/01/18 Javascript
完美解决linux下node.js全局模块找不到的情况
2018/05/16 Javascript
解决Vue axios post请求,后台获取不到数据的问题方法
2018/08/11 Javascript
Vue resource三种请求格式和万能测试地址
2018/09/26 Javascript
图文详解vue框架安装步骤
2019/02/12 Javascript
关于element-ui的隐藏组件el-scrollbar的使用
2019/05/29 Javascript
使用js实现单链解决前端队列问题的方法
2020/02/03 Javascript
Vue+axios封装请求实现前后端分离
2020/10/23 Javascript
JS画布动态实现黑客帝国背景效果
2020/11/08 Javascript
[45:15]Optic vs VP 2018国际邀请赛淘汰赛BO3 第一场 8.24
2018/08/25 DOTA
小议Python中自定义函数的可变参数的使用及注意点
2016/06/21 Python
Python爬虫实例_城市公交网络站点数据的爬取方法
2018/01/10 Python
Python 实现网页自动截图的示例讲解
2018/05/17 Python
python pprint模块中print()和pprint()两者的区别
2020/02/10 Python
英国女士和男士时尚服装网上购物:Top Labels Online
2018/03/25 全球购物
教师考核材料
2014/05/21 职场文书
公司门卫工作职责
2014/06/28 职场文书
二胎满月酒致辞
2015/07/29 职场文书
Python尝试实现蒙特卡罗模拟期权定价
2022/04/21 Python
python实现简单的三子棋游戏
2022/04/28 Python