def adjust_heap(res,start,end):

? ? '''

? ? 調(diào)整大頂堆，其中res為待排序堆列表

? ? （為了便于操作res索引，res首位補(bǔ)0，真實(shí)數(shù)據(jù)索引從1開(kāi)始）

? ? '''? ?

? ? i = start

? ? j = 2*i

? ? while j <= end:

? ? ? ? if j < end and res[j] < res[j+1]:

? ? ? ? ? ? j += 1

? ? ? ? if res[i] < res[j]:

? ? ? ? ? ? res[i],res[j] = res[j],res[i]

? ? ? ? ? ? i = j

? ? ? ? ? ? j = 2*i

? ? ? ? else:

? ? ? ? ? ? break

def swap(res,root,last):

? ? '''

? ? 交換根節(jié)點(diǎn)與尾節(jié)點(diǎn)

? ? '''

? ? res[root],res[last] = res[last],res[root]

def sort_heap(res):

? ? '''

? ? 堆排序

? ? '''

? ? res = [0] + res

? ? length = len(res) - 1

? ? last_par = length // 2? # 尋找父節(jié)點(diǎn)

? ? for i in range(last_par):

? ? ? ? adjust_heap(res,last_par - i,length)

? ? for i in range(length - 1):

? ? ? ? swap(res,1,length - i)

? ? ? ? adjust_heap(res,1,length - i - 1)

? ? return res[1:]

res = [50, 16, 30, 10, 60,? 90,? 2, 80, 70]

print(sort_heap(res))