先放在这里后面再来详解
概念
大顶堆(最大堆):要求根节点的关键字既大于或等于左子树的关键字值,又大于或等于右子树的关键字值。
最小堆则正好相反。
def heap_sort(lst):
def sift_down(start, end):
"""最大堆调整"""
root = start
while True:
child = 2 * root + 1
if child > end:
break
if child + 1 <= end and lst[child] < lst[child + 1]: # right child 大
child += 1 # child 的含义变为 right child 的位置
if lst[root] < lst[child]: # root 小于任意一个 child
lst[root], lst[child] = lst[child], lst[root] # swap
root = child #
else:
break
# 创建最大堆
for start in range((len(lst) - 2) // 2, -1, -1):
sift_down(start, len(lst) - 1)
# 堆排序
for end in range(len(lst) - 1, 0, -1):
lst[0], lst[end] = lst[end], lst[0]
sift_down(0, end - 1)
return lst
网友评论