基本算法——堆排序——python实现

目录

知识积累： 

具体： 

​ 第一步：建堆

第二步：主体

第三步：验证

第四步: 将堆释放

完整代码：

---

知识积累： 

具体： 

 第一步：建堆

具体思路看注解

我们要大根堆

大的在上面

def sift(li,low,hight):
    '''
    li ； 列表
    low ： 根节点的位置
    hight : 堆最后一个元素的位置
    return
    '''

    i = low
    j = 2*i +1
    tem = li[low]  #记录堆顶
    while j <= hight: #只要j位置有数字
        if j+1 <= hight and li[j+1] > li[j]:  #如果右孩子大于左孩子
            j = j + 1 #指向右孩子

        if li[j] > tem: #下一层大于当前的
            li[i] = li[j] #交换值
            i = j #再往下看一层
            j = 2 * i + 1 #重置j
        else:   #tem更大，把tem放在i的位置上
            li[i] = tem
            break
    else:
        li[i] = tem

第二步：主体

从下往上开始找父节点 建堆

 

def heap_sort(li):
    n = len(li)
    #建堆
    for i in range((n-2)//2,-1,-1):
        #i表示建堆的时候调整部分的下标
        sift(li,i,n-1)
        #建堆完成
    print(li)

第三步：验证

li = [i for i in range(100)]
import random
random.shuffle(li)
heap_sort(li)

结果 

第四步: 将堆释放

    for i in range(n-1,-1,-1):
        #i 指向当前堆最后一个元素
        li[0],li[i] = li[i],li[0]
        sift(li,0,i-1)#i-1是新的hight

 

完整代码：

""" 
CSDN : heart_6662
PYTHON amateur   
"""
def sift(li,low,hight):
    '''
    li ； 列表
    low ： 根节点的位置
    hight : 堆最后一个元素的位置
    return
    '''

    i = low
    j = 2*i +1
    tem = li[low]  #记录堆顶
    while j <= hight: #只要j位置有数
        if j+1 <= hight and li[j+1] > li[j]:  #如果右孩子大于左孩子
            j = j + 1 #指向右孩子

        if li[j] > tem:
            li[i] = li[j]
            i = j #往下看一层
            j = 2 * i + 1
        else:   #tem更大，把tem放在i的位置上
            li[i] = tem
            break
    else:
        li[i] = tem

def heap_sort(li):
    n = len(li)
    #建堆
    for i in range((n-2)//2,-1,-1):
        #i表示建堆的时候调整部分的下标
        sift(li,i,n-1)
        #建堆完成
        #将堆释放
    for i in range(n-1,-1,-1):
        #i 指向当前堆最后一个元素
        li[0],li[i] = li[i],li[0]
        sift(li,0,i-1)#i-1是新的hight


li = [i for i in range(100)]
import random
random.shuffle(li)
print(li)
heap_sort(li)
print(li)