【算法】Go--堆排序

package main

import "fmt"

func main() {
	nums := []int{1, 4, 0, 9, 3, 5, 8, 5, 2, 7}
	HeapSort(nums)
	fmt.Println(nums)
}
func HeapSort(nums []int) {
	BuildHeap(nums) // 构建堆
	fmt.Println(nums) // 调整堆
	for i := 0; i < len(nums)-1; i++ { // 循环次数，len(nums) - 1 
		// 堆的最后一元素位置为 len(nums) - 1 - i
		nums[0], nums[len(nums)-i-1] = nums[len(nums)-i-1], nums[0]
		// 剩余堆的大小为len(nums)-1-i，继续调整
		AdjustHeap(nums, 0, len(nums)-i-1) 
	}
}
func BuildHeap(nums []int) {
	for i := len(nums)/2 - 1; i >= 0; i-- { // 最后一个非叶子节点的index 为 len(nums)/2 - 1
		AdjustHeap(nums, i, len(nums))
	}
}
// 调整为大根堆
func AdjustHeap(nums []int, i, size int) {
	left := i*2 + 1 // 左孩子下标 ： i*2 + 1
	right := left + 1// 右孩子下标 ： i*2 + 2
	maxValIdx := i // 假设 根为 三者中的最大，依次和左孩子，右孩子比较大小
	if left < size && nums[maxValIdx] < nums[left] { 
		maxValIdx = left
	}
	if right < size && nums[maxValIdx] < nums[right] {
		maxValIdx = right
	}
	if maxValIdx != i { // 左或者右孩子的值大于根，交换位置，递归调整根的位置，maxValidx
		nums[i], nums[maxValIdx] = nums[maxValIdx], nums[i]
		AdjustHeap(nums, maxValIdx, size)
	}
}