$ 理解堆排序参考:
堆排序算法(图解详细流程) https://blog.csdn.net/u010452388/article/details/81283998
堆排序(大顶堆、小顶堆)----C语言 https://www.cnblogs.com/lanhaicode/p/10546257.html
堆排序(Heapsort)
堆积是一个近似完全二叉树的结构,并同时满足堆积的性质:即子结点的键值或索引总是小于(或者大于)它的父节点。
大顶堆:每个节点的值都大于或等于其子节点的值,在堆排序算法中用于升序排列
小顶堆:每个节点的值都小于或等于其子节点的值,在堆排序算法中用于降序排列
$ 堆排序代码参考:菜鸟教程 https://www.runoob.com/python3/python-heap-sort.html
加了一些注释,希望没有理解错误~
#大顶堆排序
#列表长度为n,以节点i及其下方的树为范围,找到最大值并换给节点i
def heapify(arr, n, i):
largest = i
l = 2 * i + 1 # 节点i如果有左子节点,它是left = 2*i + 1
r = 2 * i + 2 # 节点i如果有右子节点,它是right = 2*i + 2
if l < n and arr[i] < arr[l]: #判断i有左子节点且其值大于i节点的值,则把索引l赋值给最大值标记
largest = l
if r < n and arr[largest] < arr[r]: #判断i有左子节点且其值大于i节点的值,则把索引r赋值给最大值标记
largest = r
if largest != i: #如果最大值标记不是索引i,则交换arr[i]和arr[largest]
arr[i],arr[largest] = arr[largest],arr[i] # 交换
heapify(arr, n, largest) #从下往上调用,被交换的子节点刚才比它的字节的大,现在可不一定,所以这里要调用
def heapSort(arr):
n = len(arr)
# Build a maxheap.
for i in range(n, -1, -1): #从下往上,即从索引n到0,构建第一个大顶堆
heapify(arr, n, i)
# 一个个交换元素
for i in range(n-1, 0, -1):
arr[i], arr[0] = arr[0], arr[i] #交换,每次都把堆顶最大元素和堆底最后一个元素交换
heapify(arr, i, 0) #最后一个最大的元素,剩下的元素再次构建大堆顶
arr = [ 12, 11, 13, 5, 6, 7]
heapSort(arr)
n = len(arr)
print ("排序后")
for i in range(n):
print ("%d" %arr[i])