python中的堆

python中的堆以及堆操作

# heapqt提供了对应堆的支持，构建的是小顶堆
from heapq import *

my_data = [7,6,3,2,9,0,1,5,4]

# 对my_data应用堆属性，即堆化
heapify(my_data)
print(my_data) # [0, 2, 1, 4, 9, 3, 7, 5, 6]

# 添加一个元素，堆自动调整，继续满足小顶堆
heappush(my_data,0.5)
print(my_data) # [0, 0.5, 1, 4, 2, 3, 7, 5, 6, 9]

# 弹出堆中最小元素（即堆顶元素）
print(heappop(my_data)) # 0
# 堆自动调整
print(my_data) # [0.5, 2, 1, 4, 9, 3, 7, 5, 6]

# 弹出堆顶元素，同时压入一个元素
print(heapreplace(my_data,4.5)) # 0.5
# 将压入的元素上浮到合适位置，继续满足小顶堆
print(my_data) # [1, 2, 3, 4, 9, 4.5, 7, 5, 6]

# 返回堆中最大的n个元素
print(nlargest(3,my_data)) # [9, 7, 6]
# 返回堆中最小的n个元素
print(nsmallest(3,my_data)) # [1, 2, 3]