数据结构与算法--Python 学习笔记（三）

4、堆排序（理解不难 代码小难）

 

 

#调整函数
import random
li=[random.randint(0,60)for i in range(10)]
def pp(li,low,high):
    #li:列表
    #low:堆顶（根节点）
    #high:堆尾（最后一个元素的位置）
    i=low
    j=2*i+1
    a=li[low]#把堆顶存起来
    while j<=high:#只要j位置有数
        if j+1<=high and li[j+1]>li[j]:#如果右孩子有且大
            j=j+1#j指向右孩子
        if li[j]>a:
            li[i]=li[j]
            i=j#往下一层
            j=2*i+1
        else:#a更大,把a放到i位置上
            li[i]=a
            break
    else:
        li[i]=a
#创建堆
def hh(li):
    n=len(li)
    for i in range((n-2)//2,-1,-1):#倒着遍历到0
        #i代表建堆的时候调整的部分的下标
        pp(li,i,n-1)
        #建立完成
    for i in range(n-1,-1,-1):
        #i指向当前堆的最后一个元素
        li[0],li[i]=li[i],li[0]
        pp(li,0,i-1)#i-1是新的high

print(li)
hh(li)
print(li)