python heapq(堆)

# -*- coding: utf-8 -*-

# 堆：用数组实现的二叉树
# 并不是每一个最大堆、最小堆都是一个有序数组，要将堆转换成有序数组，需要使用堆排序

import heapq
from heapq import *

print(heapq.__all__)
my_data = list(range(10))
my_data.append(0.5)
print('my_data的元素: ', my_data)
# 对my_data列表应用堆属性 ---> 小顶堆
heapify(my_data)
print('应用堆之后my_data的元素：', my_data)

heappush(my_data, 7.2)
print('添加7.2之后my_data的元素：', my_data)  # [0, 0.5, 2, 3, 1, 5, 6, 7, 8, 9, 4, 7.2]

# 弹出堆中最小的元素
print(heappop(my_data))
print(heappop(my_data))
print('弹出两个最小元素之后my_data的元素', my_data)   # [1, 3, 2, 7, 4, 5, 6, 7.2, 8, 9]

# 弹出最小元素，压入指定元素  heapreplace(heap, x)
heapreplace(my_data, 0)
print('执行heapreplace之后my_data的元素：', my_data)   # [0, 3, 2, 7, 4, 5, 6, 7.2, 8, 9]

# 将item 入堆，然后弹出并返回堆中的最小元素
print('最小元素为：', heappushpop(my_data, -3))   # 最小元素为： -3

# my_data中最大 最小的三个元素
print('my_data中最大的三个元素：', nlargest(3, my_data))    # my_data中最大的三个元素： [9, 8, 7.2]
print('my_data中最小的三个元素：', nsmallest(3, my_data))   # my_data中最小的三个元素： [0, 2, 3]
print(my_data)

运行结果：