堆的基本定义
堆是一种树形数据结构,具有以下性质:
堆是一颗完全二叉树
堆中每个节点的值都必须大于等于或小于等于其子节点的值(最小堆和最大堆)在 Python 中,通常使用列表来表示堆。在最小堆中,列表中最小的元素是堆的根节点,也就是列表的第一个元素。在最大堆中,最大的元素是根节点,也就是列表的第一个元素。
创建堆
Python 中可以使用 heapq 模块中的 heapify() 函数来创建列表的堆。例如:
import heapq
A = [3, 1, 4, 1, 5, 9, 2, 6]
# 使用 heapify() 函数将列表转换为最小堆
heapq.heapify(A)
print(A)
# 输出:[1, 1, 2, 6, 5, 9, 4, 3]
在上述示例中,heapq.heapify(A) 将列表 A 转换为最小堆,打印结果是 [1, 1, 2, 6, 5, 9, 4, 3]。
插入元素
Python 中可以使用 heapq 模块中的 heappush() 函数将元素插入到一个堆中。例如:
import heapq
A = [1, 2, 3]
# 将元素 4 插入到堆中
heapq.heappush(A, 4)
print(A)
# 输出:[1, 2, 3, 4]
在上述示例中, heapq.heappush(A, 4) 将元素 4 插入到了列表 A 的堆中,打印的结果是 [1, 2, 3, 4]。
弹出元素
Python 中可以使用 hea