堆排序要经过两个步骤:
1.建造一个堆
2.进行堆排序
大根堆:根比左右子树的值都大(和二叉搜索树不一样)
代码解释一下:
首先要建造堆,找到最大的数放在根部,max_heap(arr,n,largest)主要就是递归,对子树进行调整。
随后到build_heap方法,主要就是先从底部到根部构建堆,在把根部最大的换到底部,再把根部以下的构建堆,进行递归。
建立大根堆算法复杂度是O(N)
对于一个索引下标为i的节点,假设它前面i-1个节点已经是大根堆,那么对i节点与根节点之间的树高度为logi,那么对i节点进行大根堆排序就是这个节点所在的层与根节点之间的树高度,即log i
所以算法复杂度为:
N=log1+log2+log3+…+logN
大根堆变为小根堆的算法复杂度是O(nlogn)
建造大根堆得递归解释很清晰:https://www.cnblogs.com/shiqi17/p/9694938.html
堆排序,首位交换,0-n-2在构建大根堆
# -*- coding:utf-8 -*-
def max_heap(arr,n,i):
largest=i
l=2*i+1
r=2*i+2
if l<n and arr[l]>=arr[largest]:
largest=l
if r<n and arr[r]>=arr[largest]:
largest=r
if largest!=i:
arr[i],arr[largest]=arr[largest],arr[i]
max_heap(arr,n,largest)
def build_heap(arr,n):
#获得一个大顶堆
for i in range((n-1)//2,-1,-1):
max_heap(arr,n,i)
for j in range(n-1,-1,-1):
arr[0],arr[j]=arr[j],arr[0]
max_heap(arr,j-1,0)