07-十大排序算法-堆排序-Python实现

算法排序 【堆排序】

时间	作者	版本	内容	备注
2023/08/03	henry_oulen@163.com	v1.0	算法排序-堆排序

1.堆排序介绍

思想准备：什么是堆？什么是大根堆和小根堆？

堆是具有以下性质的完全二叉树：父节点的数据大于子节点的数据称之为大根堆，可以用来做升序；父节点的数据小于子节点的数据称之为小根堆，可以用来做降序。对于上面的数组{4，6，8，5，9}来说可以构成如下的树：

将它构成大根堆或者小根堆则如下图所示：

基本思想：将待排序序列构成一个大顶堆，此时，整个序列的最大值就是堆顶的根节点。将其于末尾元素进行交换，此时末尾就为最大值，然后将剩余n-1个元素重新构成一个堆，这样就会得到n个元素的次小值，如此反复执行就能得到一个有序序列。

算法实现：步骤一，构造初始堆。将给定的无序序列构造成一个大顶堆

（1）初始无序序列结构如下：

（2）从最后一个非叶子节点开始，从左到右，从下到上进行调整。（第一个非叶子节点length/2 - 1 = 5/2 - 1 = 1，即下面的6节点）

（3）找到第二个非叶子节点4，因为{4,9,8}中9最大，则4和9交换

（4）这时的交换使得{4，5，6}的结构发生变化，继续调整，{4，5，6}中6最大，交换4和6

这样就将无序序列构造成了一个大根堆。

***步骤二：***将堆顶元素与数组的末尾元素交换，使得末尾元素最大，然后继续调整堆结构，再将堆顶元素与末尾元素交换，得到第二大元素，如此反复交换、调整就可以得到一个有序序列。

（1）将堆顶元素9与末尾元素4交换

（2）调整堆结构，让它继续满足堆定义

（3）将堆顶元素8与末尾元素5交换，得到第二大元素8

（4）将堆顶元素5与末尾元素4交换‘

（5）最后调整堆结构将5和6交换则得到了有序序列

2.代码实现-Python

#!/usr/bin/python3
# -*- coding:utf-8 -*-
"""
@author: henry_oulen@163.com
@file: 07-堆排序.py
@time: 2023/8/4 16:14
@desc:
"""
import random
import time


def func_time(func):
    def innrer(*args, **kwargs):
        start_time = time.time()
        result = func(*args, *kwargs)
        end_time = time.time()
        print('函数运行时间为：', end_time - start_time, 's')
        return result

    return innrer


def heapify(arr, low: int, hight: int):
    """

    :param arr: 待排序数组
    :param leftIndex: 左下标
    :param rightIndex: 右下标
    :return:
    """
    i = low
    j = low * 2 + 1
    temp = arr[i]

    while j <= hight:
        if j + 1 <= hight and arr[j] < arr[j + 1]:
            j += 1
        if arr[j] > temp:
            arr[i] = arr[j]  # 进行赋值
            i = j
            j = 2 * i + 1
        else:
            break
    arr[i] = temp


def heapSort(arr: list):
    if len(arr) <= 1:
        return arr

    n = len(arr)

    i = n // 2 - 1
    while i > -1:
        heapify(arr, i, n - 1)
        i -= 1

    # 挨个出数据

    for height in range(n - 1, -1, -1):
        arr[0], arr[height] = arr[height], arr[0]
        heapify(arr, 0, height - 1)

    return arr


# @func_time
def main(arr: list):
    """
    :param arr: 需要排序的列表 长度大于1
    :return:
    """

    arr = heapSort(arr)
    return arr


if __name__ == '__main__':
    arr = [random.randint(0, 10000) for i in range(10)]
    print("排序前列表：", arr)
    arr = main(arr)
    print("排序后列表：", arr)