选择排序——堆排序_对根据马序列进行堆排序-CSDN博客

本文链接：https://blog.csdn.net/qq_40946639/article/details/102257641

本文借鉴于lsgo实验室创始人马老师
堆排序是一种树形选择排序，是对直接选择排序的有效改进。
在这里插入图片描述

算法思想：构建大根堆之后，输出堆顶记录，对剩余的n-1个记录继续构建大根堆，便可得到n个记录的次大值，如此反复执行，就可以得到一个有序序列。这个过程叫堆排序。
在这里插入图片描述

程序：

def Restore(key,j,vCount):
    """构建以结点j为根，一共有vCount个结点的大根堆，从j结点往下遍历"""
    while j <= vCount//2:
        # 选树节点中最大的那个为根节点
        m =  2*j+1 if (2 * j +1 <= vCount and key[2 *j +1] > key[2*j]) else 2*j  # 获取子结点中较大的一个
        if key[m] > key[j]:  # 如果父节点小于子节点，那么子节点和父节点互换,则此树节点构成了大根堆
            temp = key[m]
            key[m] = key[j]
            key[j] = temp
            j = m  # 查看被替换的关键字的结点，防止替换下去的关键字导致下面的树无法构成大根堆，因此对其重新构建大根堆
        else:  # 如果父节点大于子节点，那么从此父节点往下都是大根堆（由前面的回溯可得），所有直接break出来
            break

def HeapSelectSort(nums):
    vCount = len(nums)
    tempKey = [0] + nums
    for i in range(vCount//2,0,-1):  # 从最下面的树节点往上构建大根堆
        Restore(tempKey,i, vCount)
    for i in range(vCount,1,-1):
        temp = tempKey[i]
        tempKey[i] = tempKey[1]
        tempKey[1] = temp
        Restore(tempKey,1,i-1)
    return tempKey[1:]

a = [49, 38, 65, 97, 60, 76, 13, 27, 5, 1,1]
result = HeapSelectSort(a)
print(result)