Python：堆排序

相关概念

堆是一种完全二叉树，即按从上到下，从左到右生成的二叉树。

堆有两种类型: 大根堆，小根堆。

大根堆：每个结点的值都大于或等于左右孩子结点。

小根堆：每个结点的值都小于或等于左右孩子结点。

计算某个节点的父节点和两个子节点

如上图，选取3这个节点，他的序号：i = 3

父节点：parent = ( i - 1 ) / 2 = 1 （这里进行的是取整）

左子节点：c1 = 2i + 1 = 7

右子节点：c2 = 2i + 2 = 8

堆排序的步骤

第一步：能够将某一节点及其子节点堆化（heapify）；

第二步：从最后一个父节点开始，向前对他们进行第一步操作，最终得到一个大根堆；

第三步：将根节点值（最大值）和最后一个节点值进行交换；

第四步：将除最后一个节点外的元素，进行堆化操作。

复制代码

堆化某一节点和其子节点

def heapify(tree, node, i):

# 对 i 节点进行堆化，i 是 parent 节点
# 需要找到 i 的两个子节点
c1 = 2 * i + 1
c2 = 2 * i + 2

# 需要找出 i，c1，c2 这三个节点中的最大值
# 先假设 i 为最大值
max_node = i

# 判断 c1和c2 是否存在，找到最大值的下标，即max_node
if c1 < node and tree[c1] > tree[max_node]:
    max_node = c1
if c2 < node and tree[c2] > tree[max_node]:
    max_node = c2

# max_node 不等于 i，则说明存在子节点大于 i这个父节点
# 将这个子节点的值和父节点的值进行交换，最大值成为根节点
if max_node != i:
    tree[i], tree[max_node] = tree[max_node], tree[i]
    heapify(tree, node, max_node)

堆排序

def heapSort(arr):
n = len(arr)
last_parent = (n - 2) // 2

# 从最后一个父节点开始向前进行 heapify 操作
# 将 arr 堆化，最大值成为根节点
for i in range(last_parent, -1, -1):
    heapify(arr, n, i)

for i in range(n - 1, 0, -1):
    # 将根节点和最后一个节点进行值交换
    arr[i], arr[0] = arr[0], arr[i]
    # 剩下的 i 个节点进行 heapify
    heapify(arr, i, 0)

lis = [2, 3, 5, 1, 6]
heapSort(lis)
print(lis)