堆排序(python)

標籤：

堆排序演算法

先看第一個函數HeapAdjust，這個函數功能是在已經是構成堆的二叉樹上，如果節點k資料變化了，對節點k進行修正，使之成為一個新的堆二叉樹，n為資料長度。

def HeapAdjust(lst,k,n):    while(2*k+1<n):        j=2*k+1        if j+1<n and lst[j]>lst[j+1]:            j=j+1        if lst[j]<lst[k]:            temp=lst[k]            lst[k]=lst[j]            lst[j]=temp            k=j        else:            break    return lst

HeapSort函數，利用上面的第一個函數構建堆二叉樹，並利用上面的函數進行排序

def HeapSort(lst):    n=len(lst)    for i in range(int(n/2)-1,-1,-1):        lst=HeapAdjust(lst,i,n)    print(lst)    for i in range(n-1,0,-1):        temp=lst[0]        lst[0]=lst[i]        lst[i]=temp        lst=HeapAdjust(lst,0,i)    return lst

測試函數

a=[1,5,2,8,3,4,6,9,7]print(a)result=HeapSort(a)print(result)

堆排序(python)