Python -- 堆数据结构 heapq

原创 2012年09月17日 14:56:19

import heapq

help(heapq)

heapq 是一个最小堆,堆顶元素 a[0] 永远是最小的. 和 Java 中的优先队列类似.

-------------------------------------------------

Help on module heapq:


NAME
    heapq - Heap queue algorithm (a.k.a. priority queue).


FILE
    /usr/lib64/python2.4/heapq.py


DESCRIPTION
    Heaps are arrays for which a[k] <= a[2*k+1] and a[k] <= a[2*k+2] for
    all k, counting elements from 0.  For the sake of comparison,
    non-existing elements are considered to be infinite.  The interesting
    property of a heap is that a[0] is always its smallest element.
    
    Usage:
    
    heap = []            # creates an empty heap
    heappush(heap, item) # pushes a new item on the heap
    item = heappop(heap) # pops the smallest item from the heap
    item = heap[0]       # smallest item on the heap without popping it
    heapify(x)           # transforms list into a heap, in-place, in linear time
    item = heapreplace(heap, item) # pops and returns smallest item, and adds
                                   # new item; the heap size is unchanged
    
    Our API differs from textbook heap algorithms as follows:
    
    - We use 0-based indexing.  This makes the relationship between the
      index for a node and the indexes for its children slightly less
      obvious, but is more suitable since Python uses 0-based indexing.
    
    - Our heappop() method returns the smallest item, not the largest.
    
    These two make it possible to view the heap as a regular Python list
    without surprises: heap[0] is the smallest item, and heap.sort()
    maintains the heap invariant!


FUNCTIONS
    heapify(...)
        Transform list into a heap, in-place, in O(len(heap)) time.
    
    heappop(...)
        Pop the smallest item off the heap, maintaining the heap invariant.
    
    heappush(...)
        Push item onto heap, maintaining the heap invariant.
    
    heapreplace(...)
        Pop and return the current smallest value, and add the new item.
        
        This is more efficient than heappop() followed by heappush(), and can be
        more appropriate when using a fixed-size heap.  Note that the value
        returned may be larger than item!  That constrains reasonable uses of
        this routine unless written as part of a conditional replacement:
        
                if item > heap[0]:
                    item = heapreplace(heap, item)
    
    nlargest(...)
        Find the n largest elements in a dataset.
        
        Equivalent to:  sorted(iterable, reverse=True)[:n]
    
    nsmallest(...)
        Find the n smallest elements in a dataset.
        

        Equivalent to:  sorted(iterable)[:n]

构建元素个数为 K=5 的最小堆代码实例:

#!/usr/bin/env python
# -*- encoding: utf-8 -*-
# Author: kentzhan
#

import heapq
import random

heap = []
heapq.heapify(heap)
for i in range(15):
  item = random.randint(10, 100)
  print "comeing ", item,
  if len(heap) >= 5:
    top_item = heap[0] # smallest in heap
    if top_item < item: # min heap
      top_item = heapq.heappop(heap)
      print "pop", top_item,
      heapq.heappush(heap, item)
      print "push", item,
  else:
    heapq.heappush(heap, item)
    print "push", item,
  pass
  print heap
pass
print heap

print "sort"
heap.sort()

print heap
结果:


相关文章推荐

Python——heap

通过优先队列可以构造堆,堆是一种实用的数据结构。尽管Python中没有独立的堆类型,但是包含了一些对操作函数的模块,这个模块叫heapq,主要的操作包含如下几个: heappush(heap,x)...

浅析Python heapq模块 堆数据结构

堆的定义堆是一种特殊的树形数据结构,每个节点都有一个值,通常我们所说的堆的数据结构指的是二叉树。堆的特点是根节点的值最大(或者最小),而且根节点的两个孩子也能与孩子节点组成子树,亦然称之为堆。 堆分...

利用Python实现堆排序

在这里要事先说明一下我也是新手,很多东西我了解不是很深入,写算法完全是锻炼自己逻辑能力同时顺带帮助读研的朋友么解决一些实际问题。所以很多时候考虑的东西不是很全面能请各位看到博文的大牛们指正。对于排序算...

python实现堆排序的三种方式

python实现对排序的三种方式

[从头学数学] 第256节 Python实现数据结构:堆(Heap)

剧情提要: 阿伟看到了一本比较有趣的书,是关于《计算几何》的,2008年由北清派出版。很好奇 它里面讲了些什么,就来看看啦。 正剧开始: 星历2016年09月05日 10:16:26,...

5.4 heapq--堆队列算法

本模块实现了堆队列算法,也叫作优先级队列算法。堆队列是一棵二叉树,并且拥有这样特点,它的父节点的值小于等于任何它的子节点的值,如果采用数组array实现,可以把它们的关系表示为:heap[k] 和 h...

python 算法实现 - 最小堆

最近浅学python,甚感方便,准备用python实现一遍常用的算法。 class MinHeap: def __init__(self): self.list = [0] self.si...

Python 技巧(二) 最小堆实现以及维护一个大小为100的最小堆

#!/usr/bin/env python #coding:utf-8 __author__ = 'sws'class MinHeap: ''' 最小堆 ''' ...

Delphi7高级应用开发随书源码

  • 2003年04月30日 00:00
  • 676KB
  • 下载

Python建最小堆

直接贴代码,比较好懂#_*_coding:utf_8_ ''' 建立一个最小堆 ''' import heapq import random class MyHeap(): def...
  • ustcqi
  • ustcqi
  • 2013年01月15日 13:31
  • 844
内容举报
返回顶部
收藏助手
不良信息举报
您举报文章:Python -- 堆数据结构 heapq
举报原因:
原因补充:

(最多只允许输入30个字)