heapInsert函数:将加入的节点的数与父节点的数值进行比较,若父节点的数值大于该新加入节点的数值,那么就不做变化,若父节点的数值小于该新加入节点的数值,那么就将该节点与父节点进行调换,然后继续再与下一个父节点进行比较,直到该节点成为根节点或者父节点大于该节点的情况就停。这样做的目的是为了将此完全二叉树变成一个大根堆。
// index : insert的值在arr中的索引
// 将arr 变成大根堆
func heapInsert(arr *[]int, index int) {
// 父节点为 (index - 1)/2
for (*arr)[(index-1)/2] < (*arr)[index] {
//fmt.Println("heapInsert")
moddle.Swap(arr, (index-1)/2, index)
index = (index - 1) / 2
}
heapIfy函数:根节点(或任意一节点)更换了数值,则比较其与其左右孩子中较大的那个孩子的大小,若小于则交换位置,直到遇到不小于左右孩子或者到了heapSize的位置(heapSize是该大根堆的节点总数)。
func heapIfy(arr *[]int, index int, heapSize int) {
// 首先要判断根节点是否还存在左孩子,如果存在左孩子,说明还有向下比较大小,直到变成大根堆
left := index*2 + 1
// 只要left小于heapSize,就说明index至少还存在一个孩子
for left < heapSize {
//fmt.Println("heapIfy", left, " ", heapSize, " ", index)
//moddle.PrintArr(*arr)
var largest int
// 判断左右孩子是否存在和比较左右孩子大小
if left+1 < heapSize && (*arr)[left+1] > (*arr)[left] {
largest = left + 1
} else {
largest = left
}
// 判断孩子最大值是否比index大
if (*arr)[largest] < (*arr)[index] {
break
}
moddle.Swap(arr, index, largest)
index = largest
left = index*2 + 1
}
}
heapSort函数:将数组做heapInsert,变成大根堆,然后将根节点与最后一个节点交换位置,此时数组中的一个值已经来到了正确的位置,heapSize--(代表大根堆中需要排序的数值减少了一个),然后做heapIfy(因为此时不是大根堆,做heapIfy将其再变成大根堆),然后再重复以上操作。
func HeapSort(arr *[]int) {
n := len(*arr)
for i := 0; i < n; i++ {
heapInsert(arr, i)
}
//moddle.PrintArr(*arr)
heapSize := n - 1
// 将大根堆的根节点(也就是数组的第一个数值与最后一个数交换位置)放到最后,这样就确定好了第一个最大的数
moddle.Swap(arr, 0, heapSize)
for heapSize > 0 {
heapIfy(arr, 0, heapSize)
heapSize--
moddle.Swap(arr, 0, heapSize)
}
}
时间复杂度O(NlogN)
空间复杂度O(1)