python中的堆排序peapq模块

    heapq的官方文档和源码:

    下面通过举例的方式说明heapq的应用方法

    heappop()

    heapq.heappop(heap):从堆数组heap中取出最小的值,并返回。

    1. >>> from heapq import * #引入heapq模块
    2. >>> h
    3. []
    4. >>> heappush(h,5) #向堆中依次增加数值
    5. >>> heappush(h,2)
    6. >>> heappush(h,3)
    7. >>> h #h的值
    8. [2, 5, 3, 9]
    9. >>> heappop(h) #从h中删除最小的,并返回该值
    10. 2
    11. >>> h
    12. >>> h.append(1) #注意,如果不是压入堆中,而是通过append追加一个数值
    13. >>> h #堆的函数并不能操作这个增加的数值,或者说它堆对来讲是不存在的
    14. [3, 5, 9, 1]
    15. >>> heappop(h) #从h中能够找到的最小值是3,而不是1
    16. 3
    17. >>> heappush(h,2) #这时,不仅将2压入到堆内,而且1也进入了堆。
    18. >>> h
    19. [1, 2, 9, 5]
    20. >>> heappop(h) #操作对象已经包含了1

    是上述heappush和heappop的合体,同时完成两者的功能.注意:相当于先操作了heappush(heap,item),然后操作heappop(heap)

    heapq.heapify(x)

    1. >>> a=[3,6,1]
    2. >>> heapify(a) #将a变成堆之后,可以对其操作
    3. 1
    4. >>> b=[4,2,5] #b不是堆,如果对其进行操作,显示结果如下
    5. >>> heappop(b) #按照顺序,删除第一个数值并返回,不会从中挑选出最小的
    6. 4
    7. >>> heapify(b) #变成堆之后,再操作
    8. >>> heappop(b)
    9. 2

    是heappop(heap)和heappush(heap,item)的联合操作。注意,与heappushpop(heap,item)的区别在于,顺序不同,这里是先进行删除,后压入堆

    heapq.merge(*iterables)

    举例:

    1. >>> a=[2,4,6]
    2. >>> b=[1,3,5]
    3. >>> c=merge(a,b)
    4. >>> list(c)
    5. [1, 2, 3, 4, 5, 6]

    获取列表中最大、最小的几个值。