go语言实现堆排序

//建大根堆
func HeapAdjust(arr []int, s, m int) {
    //该函数表示数组arr[s:m]段除了arr[s]其余部分都满足堆的情况，
	rc := arr[s]
	for j := 2*s + 1; j < m; j = j*2 + 1 {
        //2*s+1 表示左子树，
		if j < m && arr[j] < arr[j+1] { //左子树的值小于右子树的值
			j++ //j++表示右子树
		}
		if rc >= arr[j] { //大于，表示比左右子树的最大值还要大，满足条件跳出
			break
		}
		arr[s] = arr[j] //找到了大的值，进行赋值
		s = j  //s 设置为j的位置进行新一轮选择
	}
	arr[s] = rc
}

func HeapSort(arr []int) {
	for i := len(arr)/2 - 1; i >= 0; i-- { //从有子树的最大节点开始，一直到0 依次建堆
 		HeapAdjust(arr, i, len(arr))
 	}
	for i := len(arr) - 1; i > 0; i-- { 
		arr[0], arr[i] = arr[i], arr[0] //将最后一个和第一个交换再建堆
		HeapAdjust(arr, 0, i-1)  //下一次则不带最后一个元素
 	}
}

把堆看做一维数组，则可以看成是一个完全二叉树，完全二叉树所有非终端节点的值均不小于其左右孩子结点的值，由此堆顶元素则为最大值

测试代码

//heap_test.go
import (
	"fmt"
	"testing"
)

func TestSort(t *testing.T) {
	arr := []int{0, 2, 3, 4, 5, 6, 1, 0, 3}
	HeapSort(arr)
	fmt.Println(arr)
}