python 堆和优先队列的使用详解


Posted in Python onMarch 05, 2019

1.heapq

python里面的堆是通过在列表中维护堆的性质实现的。这一点与C++中heap一系列的算法类似,底层是通过堆vector的维护获取堆的性质。

关于二叉树

二叉树的特点:

二叉树是一种存储数据元素的汇集数据结构。

二叉树最重要的性质就是树的高度和树中可以容纳的最大结点个数之间的关系。树的高度类似于表长,是从根结点到其他结点的最大距离。在长为n的表里只能容纳n个结点,而在高为h的二叉树中则可以容纳大约2^h个结点,这是表和树的最大不同点。

一般的元素插入,如果是按线性顺序排列的,那么操作必然需要O(n)的时间(需要对n个数据进行移位处理),要突破这个限制,必须考虑其他数据结构的组织方式。二叉树就是一种高效插入的存储方式。

堆排序利用的是完全二叉树。

python堆的部分API,其他API查阅文档python_heap_API和  heapq的源代码

import heapq
#向堆中插入元素,heapq会维护列表heap中的元素保持堆的性质
heapq.heappush(heap, item)
#heapq把列表x转换成堆
heapq.heapify(x)
#从可迭代的迭代器中返回最大的n个数,可以指定比较的key
heapq.nlargest(n, iterable[, key])
#从可迭代的迭代器中返回最小的n个数,可以指定比较的key
heapq.nsmallest(n, iterable[, key])
#从堆中删除元素,返回值是堆中最小或者最大的元素
heapq.heappop(heap)

1.1.内置类型

从上述源代码可以看出来,heapq使用的内置的小于号,或者类的__lt__比较运算来进行比较。

def heapq_int():
  heap = []
  #以堆的形式插入堆
  heapq.heappush(heap,10)
  heapq.heappush(heap,1)
  heapq.heappush(heap,10/2)
  [heapq.heappush(heap,i) for i in range(10)]
  [heapq.heappush(heap,10 - i) for i in range(10)]
  #最大的10个元素
  print heapq.nlargest(10,heap)
  #输出所有元素
  print [heapq.heappop(heap) for i in range(len(heap))]

1.2.元组类型

元素会默认调用内置比较函数cmp

def heapq_tuple():
  heap = []
  #向推中插入元组
  heapq.heappush(heap,(10,'ten'))
  heapq.heappush(heap,(1,'one'))
  heapq.heappush(heap,(10/2,'five'))
  while heap:
    print heapq.heappop(heap),
  print

1.2.类类型

类类型,使用的是小于号_lt_,当然没有重写但是有其他的比较函数例如:_le_,_gt_,_cmp_,也是会调用的,和小于号等价的都可以调用(测试了gt),具体的这些操作之间的关系我也没有研究过。如果类里面没有重写_lt_,会调用其他的比较操作符,从源代码可以看出来,如果没有_lt_,那么会调用_ge_函数。

所以可以重写上述的那些函数:

class Skill(object):
  def __init__(self,priority,description):
    self.priority = priority
    self.description = description
  def __lt__(self,other):#operator < 
    return self.priority < other.priority
  def __ge__(self,other):#oprator >=
    return self.priority >= other.priority
  def __le__(self,other):#oprator <=
    return self.priority <= other.priority
  def __cmp__(self,other):
    #call global(builtin) function cmp for int
    return cmp(self.priority,other.priority)
  def __str__(self):
    return '(' + str(self.priority)+',\'' + self.description + '\')'

def heapq_class():
  heap = []
  heapq.heappush(heap,Skill(5,'proficient'))
  heapq.heappush(heap,Skill(10,'expert'))
  heapq.heappush(heap,Skill(1,'novice'))
  while heap:
    print heapq.heappop(heap),
  print

所以如果要用到自己定义的类型,可以重写上述函数,就可以使用heapq函数了。

2.PriorityQueue

PriorityQueue的python源代码PriorityQueue 

从源代码可以看出来,PriorityQueue使用的就是heapq来实现的,所以可以认为两者算法本质上是一样的。当然PriorityQueue考虑到了线程安全的问题。

下面给出PriorityQueue的部分API和使用方法。

参考Queue

#向队列中添加元素
Queue.put(item[, block[, timeout]])
#从队列中获取元素
Queue.get([block[, timeout]])
#队列判空
Queue.empty()
#队列大小
Queue.qsize()

2.1.内置类型

直接调用内置函数cmp进行比较

try:
  import Queue as Q #python version < 3.0
except ImportError:
  import queue as Q #python3.*
def PriorityQueue_int():
  que = Q.PriorityQueue()
  que.put(10)
  que.put(1)
  que.put(5)
  while not que.empty():
    print que.get(),
  print

2.2.元组类型

def PriorityQueue_tuple():
  que = Q.PriorityQueue()
  que.put((10,'ten'))
  que.put((1,'one'))
  que.put((10/2,'five'))
  while not que.empty():
    print que.get(),
  print

2.2.自定义类型

class Skill(object):
  def __init__(self,priority,description):
    self.priority = priority
    self.description = description
  #下面两个方法重写一个就可以了
  def __lt__(self,other):#operator < 
    return self.priority < other.priority
  def __cmp__(self,other):
    #call global(builtin) function cmp for int
    return cmp(self.priority,other.priority)
  def __str__(self):
    return '(' + str(self.priority)+',\'' + self.description + '\')'

def PriorityQueue_class():
  que = Q.PriorityQueue()
  skill5 = Skill(5,'proficient')
  skill6 = Skill(6,'proficient6')
  que.put(skill6)
  que.put(Skill(5,'proficient'))
  que.put(Skill(10,'expert'))
  que.put(Skill(1,'novice'))
  while not que.empty():
    print que.get(),
  print

其他的一些方法的使用还是需要参考给出的文档的。

最后一点,让我比较奇怪的是(可能我并没有找到),没有提供像排序函数那样,指定比较方法函数,这点和c++有点区别。

这篇文档参考:参考文档

以上就是本文的全部内容,希望对大家的学习有所帮助,也希望大家多多支持三水点靠木。

Python 相关文章推荐
浅谈python import引入不同路径下的模块
Jul 11 Python
简单实现python画圆功能
Jan 25 Python
终端命令查看TensorFlow版本号及路径的方法
Jun 13 Python
Python网页正文转换语音文件的操作方法
Dec 09 Python
Python实现的文轩网爬虫完整示例
May 16 Python
python获取当前文件路径以及父文件路径的方法
Jul 10 Python
利用pytorch实现对CIFAR-10数据集的分类
Jan 14 Python
python实现信号时域统计特征提取代码
Feb 26 Python
django执行数据库查询之后实现返回的结果集转json
Mar 31 Python
IDLE下Python文件编辑和运行操作
Apr 25 Python
基于Python实现视频的人脸融合功能
Jun 12 Python
Selenium浏览器自动化如何上传文件
Apr 06 Python
Python两个字典键同值相加的几种方法
Mar 05 #Python
详解python算法之冒泡排序
Mar 05 #Python
Python字符串通过'+'和join函数拼接新字符串的性能测试比较
Mar 05 #Python
Python实现KNN(K-近邻)算法的示例代码
Mar 05 #Python
Python按钮的响应事件详解
Mar 04 #Python
Python中三元表达式的几种写法介绍
Mar 04 #Python
Python生成器的使用方法和示例代码
Mar 04 #Python
You might like
PHP 中的批处理的实现
2007/06/14 PHP
Thinkphp的volist标签嵌套循环使用教程
2014/07/08 PHP
PHP中使用array函数新建一个数组
2015/11/19 PHP
PHP内存缓存功能memcached示例
2016/10/19 PHP
php实现的双色球算法示例
2017/06/20 PHP
javascript 用记忆函数快速计算递归函数
2010/03/15 Javascript
jquery实现浮动的侧栏实例
2015/06/25 Javascript
javascript使用Promise对象实现异步编程
2016/03/01 Javascript
jQuery页面元素动态添加后绑定事件丢失方法,非 live
2016/06/16 Javascript
JQuery中Ajax的操作完整例子
2017/03/07 Javascript
使用Angular Cli如何创建Angular私有库详解
2019/01/30 Javascript
JS组件库AlloyTouch实现图片轮播过程解析
2020/05/29 Javascript
原生JS实现pc端轮播图效果
2020/12/21 Javascript
Python 列表list使用介绍
2014/11/30 Python
简单介绍Python中的decode()方法的使用
2015/05/18 Python
python读取excel表格生成erlang数据
2017/08/26 Python
Python实现学校管理系统
2018/01/11 Python
Python实现替换文件中指定内容的方法
2018/03/19 Python
Python装饰器用法实例总结
2018/05/26 Python
Python实现合并两个列表的方法分析
2018/05/28 Python
python skimage 连通性区域检测方法
2018/06/21 Python
python字符串替换第一个字符串的方法
2019/06/26 Python
Pytorch自己加载单通道图片用作数据集训练的实例
2020/01/18 Python
pyecharts在数据可视化中的应用详解
2020/06/08 Python
Anaconda+spyder+pycharm的pytorch配置详解(GPU)
2020/10/18 Python
在 Python 中使用 7zip 备份文件的操作
2020/12/11 Python
python re.match()用法相关示例
2021/01/27 Python
使用canvas绘制超炫时钟
2014/12/17 HTML / CSS
基于MUI框架使用HTML5实现的二维码扫描功能
2018/03/01 HTML / CSS
巴西体育用品商店:Lojão dos Esportes
2018/07/21 全球购物
职工代表大会主持词
2014/04/01 职场文书
本科毕业生求职自荐信
2014/04/09 职场文书
2013年最新自荐信范文
2014/06/23 职场文书
建筑施工安全生产责任书
2014/07/22 职场文书
人力资源管理毕业求职信
2014/08/05 职场文书
Python读取和写入Excel数据
2022/04/20 Python