44 lines
1.8 KiB
Go
44 lines
1.8 KiB
Go
|
/*
|
||
|
|
* @Author : huangzj
|
||
|
|
* @Time : 2020/8/20 16:13
|
||
|
|
* @Description：堆结构的基本特性：
|
||
|
|
* i 结点的父结点下标就为(i – 1) / 2。
|
||
|
|
* i 节点的左右子结点下标分别为2 * i + 1和2 * i + 2。
|
||
|
|
* 具体图解可参考：https://www.cnblogs.com/chengxiao/p/6129630.html
|
||
|
|
*/
|
||
|
|
|
||
|
|
package sort
|
||
|
|
|
||
|
|
func HeapSort(list []int) {
|
||
|
|
//构建堆,构建堆的算法，会根据父节点去构建子节点，并且再往上构建父节点的父节点..所以这边的构建只要到数组长度/2 -1就行.
|
||
|
|
//这样就可以涵盖它的两个子节点.
|
||
|
|
for i := len(list)/2 - 1; i >= 0; i-- {
|
||
|
|
makeHeap(list, i, len(list))
|
||
|
|
}
|
||
|
|
|
||
|
|
//构建的是最大堆，每次最大的元素都在第0个位置，所有每一次构建堆，都要把第0个位置的数置换到对应的当前的最后一个位置.
|
||
|
|
//上面已经构建了一个完整的堆，所以这边只需要对堆进行调整即可
|
||
|
|
for j := len(list) - 1; j > 0; j-- {
|
||
|
|
list[j], list[0] = list[0], list[j] //将堆顶元素与末尾元素进行交换
|
||
|
|
makeHeap(list, 0, j) //重新对堆进行调整(这个时候是对除去了上次的最大节点的堆进行重组)
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
//构建堆算法
|
||
|
|
func makeHeap(list []int, start int, end int) {
|
||
|
|
temp := list[start] //先把当前父节点的值保存下来
|
||
|
|
for i := 2*start + 1; i < end; i = 2*start + 1 {
|
||
|
|
//如果左子树值比右子树小，那么要比较的就是右子树和父节点
|
||
|
|
if i < end-1 && list[i] < list[i+1] {
|
||
|
|
i++
|
||
|
|
}
|
||
|
|
//如果被比较的子节点更大，则应该放在父节点的位置.并且开始节点往后移动，比较下一个节点(当前父节点的子节点)的父子节点值.
|
||
|
|
if list[i] > temp {
|
||
|
|
list[start], list[i] = list[i], list[start]
|
||
|
|
start = i
|
||
|
|
} else {
|
||
|
|
break //如果父节点大于两个自己点，则说明是完整的堆，则退出循环
|
||
|
|
}
|
||
|
|
}
|
||
|
|
}
|