9种排序算法---堆排序算法

望舒:-

于 2024-09-02 05:47:59 发布

阅读量355

点赞数 9

文章标签：排序算法数据结构算法 python

本文链接：https://blog.csdn.net/2301_78576595/article/details/141793507

版权

堆的数据结构近似一个完全二叉树结构，同时满足堆积的性质，即子节点的键值或索引总是大于或小于它的父节点。所以堆必须满足两个条件：

（1）必须是一个完全二叉树结构

（2）子节点的键值或索引总是大于或小于它的父节点

首先来简单介绍一下什么是完全二叉树结构；完全二叉树结构的每个节点都有只是两条连接线，从上到下，从左到右依次生成如图1；如果我们想添加一个节点，只能添加到F节点之下，如果添加到G节点之下将不满足全二叉树结构的特点（即从上到下，从左到右依次生成）

图1

再来看看子节点的键值或索引总是大于或小于它的父节点这一个条件；图2满足条件（1）所以它是一个完全二叉树;9做为父节点大于6,5；6做为父节点大于5,2；5做为父节点大于1；因此图2同时满足条件（1）条件（2），这种结构就是堆

具体代码如下：

def heapify(arr, n, i):
    # 初始化最大值为根节点
    largest = i
    left = 2 * i + 1 # 计算左节点位置
    right = 2 * i + 2 # 计算右节点位置
    # 如果左子节点大于根节点，则更新最大值
    if left < n and arr[i] < arr[left]:
        largest = left
    # 如果右子节点大于根节点，则更新最大值
    if right < n and arr[largest] < arr[right]:
        largest = right
    # 如果最大值不是根节点，则交换根节点和最大值，并递归调整堆
    if largest != i:
        arr[i], arr[largest] = arr[largest], arr[i]
        heapify(arr, n, largest)
def heap_sort(arr):
    n = len(arr)
    # 构建大顶堆
    for i in range(n // 2 - 1, -1, -1):
        heapify(arr, n, i)
    # 依次取出堆顶元素，将其放置在序列的末尾
    for i in range(n - 1, 0, -1):
        arr[i], arr[0] = arr[0], arr[i]
        heapify(arr, i, 0)
    return arr


print(heap_sort([5, 2, 9, 1, 5, 6]))
# 输出 [1, 2, 5, 5, 6, 9]

代码解释：

（1）有这样一组数列，现在我们对其进行递增排序