博客
关于我
【python刷题】二叉堆-优先级队列
阅读量:470 次
发布时间:2019-03-06

本文共 2391 字,大约阅读时间需要 7 分钟。

我们要明确以下几点:

1、二叉堆是一棵完全二叉树;
2、可以构造大顶堆和小顶堆;
3、二叉堆构建优先级队列,以大顶堆为例,每次出队列的值都是当前队列中的最大值;

class MaxPQ:    def __init__(self):        self.n = 0  # 当前优先级队列中的元素,优先级队列:插入或删除元素的时候,元素会自动排序        self.pq = [0]  # 存储数组,0索引位置不用    def parent(self, root):        return root // 2    def left(self, root):        return root * 2    def right(self, root):        return root * 2 + 1    def getMax(self):        return self.pq[1]    def insert(self, x):        self.n += 1        self.pq.append(x)        self.up(self.n)    def delMax(self):        m = self.pq[1]        self.swap(1, self.n)        self.pq.pop()        self.n -= 1        self.down(1)        return m    def maxChildInd(self, x):        # 如果右孩子为空,返回左孩子的索引        if self.right(x) > self.n:            return self.left(x)        else:            if self.less(self.left(x), self.right(x)):                return self.right(x)            else:                return self.left(x)    # 上浮第x个元素    def up(self, x):        while self.parent(x) > 0:            # 如果新节点小于父节点            if self.pq[x] > self.pq[self.parent(x)]:                self.swap(self.parent(x), x)            x = self.parent(x)    # 下浮第x个元素    def down(self, x):        while self.left(x) <= self.n:            # 返回值较大的那个的索引            tmp = self.maxChildInd(x)            if self.pq[tmp] > self.pq[x]:                self.swap(tmp, x)            x = tmp    # 交换两个元素    def swap(self, i, j):        self.pq[i],self.pq[j] = self.pq[j],self.pq[i]    # pq[i]是否比pq[j]小    def less(self, i, j):        return self.pq[i] < self.pq[j]    def buildHeap(self, alist):        self.pq = self.pq + alist        self.n = len(alist)        ind = self.n // 2        while ind > 0:            self.down(ind)            ind -= 1maxPQ = MaxPQ()maxPQ.buildHeap([78, 83, 82, 80, 79, 65, 84])print("初始的pq:", maxPQ.pq)print(maxPQ.delMax())print(maxPQ.pq)print(maxPQ.delMax())print(maxPQ.pq)print(maxPQ.delMax())print(maxPQ.pq)print(maxPQ.delMax())print(maxPQ.pq)print(maxPQ.delMax())print(maxPQ.pq)maxPQ.insert(67)print("插入67之后的pq:", maxPQ.pq)print(maxPQ.delMax())print(maxPQ.pq)maxPQ.insert(66)print("插入66之后的pq:", maxPQ.pq)print(maxPQ.delMax())print(maxPQ.pq)

结果:

初始的pq: [0, 84, 83, 82, 80, 79, 65, 78]
84
[0, 83, 80, 82, 78, 79, 65]
83
[0, 82, 80, 65, 78, 79]
82
[0, 80, 79, 65, 78]
80
[0, 79, 78, 65]
79
[0, 78, 65]
插入67之后的pq: [0, 78, 65, 67]
78
[0, 67, 65]
插入66之后的pq: [0, 67, 65, 66]
67
[0, 66, 65]

转载地址:http://xypbz.baihongyu.com/

你可能感兴趣的文章
Mysql学习总结(69)——Mysql EXPLAIN 命令使用总结
查看>>
Mysql学习总结(6)——MySql之ALTER命令用法详细解读
查看>>
Mysql学习总结(70)——MySQL 优化实施方案
查看>>
Mysql学习总结(71)——MySQL 重复记录查询与删除总结
查看>>
Mysql学习总结(71)——数据库介绍(MySQL安装 体系结构、基本管理)再回顾
查看>>
Mysql学习总结(72)——MySQL 开发者开发,设计规范再总结
查看>>
Mysql学习总结(73)——MySQL 查询A表存在B表不存在的数据SQL总结
查看>>
Mysql学习总结(74)——慢SQL!压垮团队的最后一根稻草!
查看>>
Mysql学习总结(75)——并发量大、数据量大的互联网业务数据库设计军规
查看>>
Mysql学习总结(76)——MySQL执行计划(explain)结果含义总结
查看>>
Mysql学习总结(77)——温故Mysql数据库开发核心原则与规范
查看>>
Mysql学习总结(78)——MySQL各版本差异整理
查看>>
Mysql学习总结(79)——MySQL常用函数总结
查看>>
Mysql学习总结(7)——MySql索引原理与使用大全
查看>>
Mysql学习总结(80)——统计数据库的总记录数和库中各个表的数据量
查看>>
Mysql学习总结(81)——为什么MySQL不推荐使用uuid或者雪花id作为主键?
查看>>
Mysql学习总结(82)——MySQL逻辑删除与数据库唯一性约束如何解决?
查看>>
Mysql学习总结(83)——常用的几种分布式锁:ZK分布式锁、Redis分布式锁、数据库分布式锁、基于JDK的分布式锁方案对比总结
查看>>
Mysql学习总结(84)—— Mysql的主从复制延迟问题总结
查看>>
Mysql学习总结(85)——开发人员最应该明白的数据库设计原则
查看>>