堆排序的Python代码

最新推荐文章于 2022-08-27 08:39:45 发布

沙丁鱼鱼鱼

最新推荐文章于 2022-08-27 08:39:45 发布

阅读量437

点赞数

分类专栏：机器学习文章标签：堆排序

本文链接：https://blog.csdn.net/u014694994/article/details/94973491

版权

机器学习专栏收录该内容

14 篇文章 1 订阅

订阅专栏

# -*- coding: utf-8 -*-
"""
Created on Fri Jul  5 10:13:16 2019

@author: Yuki
"""

def heapAdjust(nums, Len, index):
    left = index*2 + 1
    right = index*2 + 2
    
    maxIndex = index
    if left<Len and nums[left] > nums[maxIndex]:
        maxIndex = left
    if right < Len and nums[right]>nums[maxIndex]:
        maxIndex = right
    
    if maxIndex != index:
        nums[index], nums[maxIndex] = nums[maxIndex], nums[index]
        heapAdjust(nums, Len, maxIndex)
      
def heapSort(nums):
    Len = len(nums)
    #构建大根堆（从最后一个非叶子节点向上）
    for i in range(Len//2-1, -1, -1):
        heapAdjust(nums, Len, i)
    
    for i in range(len(nums)-1, 0, -1):
        nums[0], nums[i] = nums[i], nums[0] #  将当前最大的放置到数组末尾
        heapAdjust(nums, i, 0)             # 将未完成排序的部分继续进行堆排序
    return nums

nums = [100,20,5,40,888,432,-100,-5]

res = heapSort(nums)