第二章:数据结构-heapq:堆排序算法-访问堆元素

2.4.3 访问堆元素
一旦堆已经被正确组织,则可以使用heappop()删除有最小值的元素。

import heapq
from heapq_showtree import show_tree
from heapq_heapdata import data

print('random    :',data)
heapq.heapify(data)
print('heapified :')
show_tree(data)
print

for i in range(2):
    smallest = heapq.heappop(data)
    print('pop   {:>3}'.format(smallest))
    show_tree(data)

这个例子是由标准库文档改写的,其中使用heapify()和heappop()对一个数字列表进行排序。
运行结果:
在这里插入图片描述
如果希望在一个操作中删除现有元素并替换为新值,则可以使用heapreplace()。

import heapq
from heapq_showtree import show_tree
from heapq_heapdata import data

heapq.heapify(data)
print('start:')
show_tree(data)

for n in [0,13]:
    smallest = heapq.heapreplace(data,n)
    print('replace {:>2} with {:>2}:'.format(smallest,n))
    show_tree(data)

通过原地替换元素,这样可以维持一个固定大小的堆,如按优先级排序的作业队列。
运行结果:
在这里插入图片描述

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值