堆排序代码实现

最新推荐文章于 2022-10-22 17:45:11 发布

静默安然

最新推荐文章于 2022-10-22 17:45:11 发布

阅读量131

点赞数

分类专栏：算法分析与设计

本文链接：https://blog.csdn.net/zhao2chen3/article/details/118813216

版权

算法分析与设计专栏收录该内容

12 篇文章 0 订阅

订阅专栏

本文介绍了如何将大根堆转换为小根堆，通过修改`keep_max_heap`函数中的比较条件实现。堆排序算法的实现过程也被详细阐述，包括初始建堆、元素交换和堆调整等步骤。示例中，给定数组`a`经过堆排序后变为降序排列。

摘要由CSDN通过智能技术生成

大根堆是升序

def keep_max_heap(h,n,root):
    left=2*root+1#下标从0开始
    right=2*root+2
    maxnum=root
    if left<n and h[left]>h[maxnum]:
        maxnum=left
    if right<n and h[right]>h[maxnum]:
        maxnum=right
    if maxnum!=root:
        h[maxnum],h[root]=h[root],h[maxnum]
        keep_max_heap(h,n,maxnum)


def heapSort(arr):
    n = len(arr)

    # 初始建大根堆  从最后一个非叶节点开始，向根节点调整
    for i in range(n//2-1, -1, -1):
        keep_max_heap(arr, n, i)


    for i in range(n - 1, 0, -1):
        # 将根的最大元素与最后一个元素交换 然后从根开始向下调整堆
        arr[i], arr[0] = arr[0], arr[i]
        # 后面是有序数组 前面是无序数组 只需调整前面未排序的数组 因此需要修改无序数组尾下标i
        keep_max_heap(arr, i, 0)
        #arr是整个数组  i是无序数组的结尾下标 0是跟下标

if __name__=='__main__':
    a=[6,5,5,4,3,3,2,2,1]
    heapSort(a)
    print(a)

变成小根堆（降序），只需蓝色位置修改成下面符号即可