python计算最大优先级队列实例


Posted in Python onDecember 18, 2013
# -*- coding: utf-8 -*-
class Heap(object):
    @classmethod
    def parent(cls, i):
        """父结点下标"""
        return int((i - 1) >> 1);
    @classmethod
    def left(cls, i):
        """左儿子下标"""
        return (i << 1) + 1;
    @classmethod
    def right(cls, i):
        """右儿子下标"""
        return (i << 1) + 2;
class MaxPriorityQueue(list, Heap):
    @classmethod
    def max_heapify(cls, A, i, heap_size):
        """最大堆化A[i]为根的子树"""
        l, r = cls.left(i), cls.right(i)
        if l < heap_size and A[l] > A[i]:
            largest = l
        else:
            largest = i
        if r < heap_size and A[r] > A[largest]:
            largest = r
        if largest != i:
            A[i], A[largest] = A[largest], A[i]
            cls.max_heapify(A, largest, heap_size)
    def maximum(self):
        """返回最大元素,伪码如下:
        HEAP-MAXIMUM(S)
        1  return A[1]
        T(n) = O(1)
        """
        return self[0]
    def extract_max(self):
        """去除并返回最大元素,伪码如下:
        HEAP-EXTRACT-MAX(A)
        1  if heap-size[A] < 1
        2    then error "heap underflow"
        3  max ← A[1]
        4  A[1] ← A[heap-size[A]] // 尾元素放到第一位
        5  heap-size[A] ← heap-size[A] - 1 // 减小heap-size[A]
        6  MAX-HEAPIFY(A, 1) // 保持最大堆性质
        7  return max
        T(n) = θ(lgn)
        """
        heap_size = len(self)
        assert heap_size > 0, "heap underflow"
        val = self[0]
        tail = heap_size - 1
        self[0] = self[tail]
        self.max_heapify(self, 0, tail)
        self.pop(tail)
        return val
    def increase_key(self, i, key):
        """将i处的值增加到key,伪码如下:
        HEAP-INCREASE-KEY(A, i, key)
        1  if key < A[i]
        2    the error "new key is smaller than current key"
        3  A[i] ← key
        4  while i > 1 and A[PARENT(i)] < A[i] // 不是根结点且父结点更小时
        5    do exchange A[i] ↔ A[PARENT(i)] // 交换两元素
        6       i ← PARENT(i) // 指向父结点位置
        T(n) = θ(lgn)
        """
        val = self[i]
        assert key >= val, "new key is smaller than current key"
        self[i] = key
        parent = self.parent
        while i > 0 and self[parent(i)] < self[i]:
            self[i], self[parent(i)] = self[parent(i)], self[i]
            i = parent(i)
    def insert(self, key):
        """将key插入A,伪码如下:
        MAX-HEAP-INSERT(A, key)
        1  heap-size[A] ← heap-size[A] + 1 // 对元素个数增加
        2  A[heap-size[A]] ← -∞ // 初始新增加元素为-∞
        3  HEAP-INCREASE-KEY(A, heap-size[A], key) // 将新增元素增加到key
        T(n) = θ(lgn)
        """
        self.append(float('-inf'))
        self.increase_key(len(self) - 1, key)
if __name__ == '__main__':
    import random
    keys = range(10)
    random.shuffle(keys)
    print(keys)
    queue = MaxPriorityQueue() # 插入方式建最大堆
    for i in keys:
        queue.insert(i)
    print(queue)
    print('*' * 30)
    for i in range(len(keys)):
        val = i % 3
        if val == 0:
            val = queue.extract_max() # 去除并返回最大元素
        elif val == 1:
            val = queue.maximum() # 返回最大元素
        else:
            val = queue[1] + 10
            queue.increase_key(1, val) # queue[1]增加10
        print(queue, val)
    print([queue.extract_max() for i in range(len(queue))])
Python 相关文章推荐
python进阶教程之动态类型详解
Aug 30 Python
轻松掌握python设计模式之策略模式
Nov 18 Python
Python中二维列表如何获取子区域元素的组成
Jan 19 Python
Python统计单词出现的次数
Apr 04 Python
pandas筛选某列出现编码错误的解决方法
Nov 07 Python
python自制包并用pip免提交到pypi仅安装到本机【推荐】
Jun 03 Python
pytorch索引查找 index_select的例子
Aug 18 Python
python 初始化一个定长的数组实例
Dec 02 Python
深入浅析python的第三方库pandas
Feb 13 Python
Python打包工具PyInstaller的安装与pycharm配置支持PyInstaller详细方法
Feb 27 Python
Python如何实现后端自定义认证并实现多条件登陆
Jun 22 Python
python关于集合的知识案例详解
May 30 Python
python计算最小优先级队列代码分享
Dec 18 #Python
python查找第k小元素代码分享
Dec 18 #Python
python获取beautifulphoto随机某图片代码实例
Dec 18 #Python
python使用urllib2模块获取gravatar头像实例
Dec 18 #Python
python2.7删除文件夹和删除文件代码实例
Dec 18 #Python
python使用xmlrpc实例讲解
Dec 17 #Python
python三元运算符实现方法
Dec 17 #Python
You might like
php checkbox复选框值的获取与checkbox默认值输出方法
2010/05/15 PHP
CodeIgniter上传图片成功的全部过程分享
2013/08/12 PHP
PHP的PSR规范中文版
2013/09/28 PHP
PHP获取当前页面URL函数实例
2014/10/22 PHP
php编写的一个E-mail验证类
2015/03/25 PHP
PHP实现单例模式建立数据库连接的方法分析
2020/02/11 PHP
表格 隔行换色升级版
2009/11/07 Javascript
基于jquery的tab切换 js原理
2010/04/01 Javascript
document.addEventListener使用介绍
2014/03/07 Javascript
JavaScript用Number方法实现string转int
2014/05/13 Javascript
jQuery实现鼠标滚轮动态改变样式或效果
2015/01/05 Javascript
jquery $(document).ready()和window.onload的区别浅析
2015/02/04 Javascript
jQuery层动画定位滑动效果的方法
2015/04/30 Javascript
浅谈javascript中的DOM方法
2015/07/16 Javascript
jquery验证手机号是否正确实例讲解
2015/11/17 Javascript
js实现模拟银行卡账号输入显示效果
2015/11/18 Javascript
本地搭建微信小程序服务器的实现方法
2017/10/27 Javascript
Vue 全家桶实现移动端酷狗音乐功能
2018/11/16 Javascript
Angular6 Filter实现页面搜索的示例代码
2018/12/02 Javascript
axios携带cookie配置详解(axios+koa)
2018/12/28 Javascript
Vue实现table上下移动功能示例
2019/02/21 Javascript
浅谈Layui的eleTree树式选择器使用方法
2019/09/25 Javascript
微信公众号网页分享功能开发的示例代码
2020/05/27 Javascript
Python3基础教程之递归函数简单示例
2019/06/07 Python
pyqt5 实现 下拉菜单 + 打开文件的示例代码
2019/06/20 Python
python实现翻转棋游戏(othello)
2019/07/29 Python
Python传递参数的多种方式(小结)
2019/09/18 Python
Python numpy线性代数用法实例解析
2019/11/15 Python
Python迭代器模块itertools使用原理解析
2019/12/11 Python
浅谈Python线程的同步互斥与死锁
2020/03/22 Python
html5中canvas图表实现柱状图的示例
2017/11/13 HTML / CSS
Office DEPOT法国官网:欧迪办公用品采购
2018/01/03 全球购物
Cynthia Rowley官网:全球领先的生活方式品牌
2020/10/27 全球购物
高级工程师岗位职责
2013/12/15 职场文书
Python中的变量与常量
2021/11/11 Python
Windows server 2012搭建FTP服务器
2022/04/29 Servers