func buildHeep(nums []int, len int) {
	// 找到最后一個節(jié)點(diǎn)的父節(jié)點(diǎn)
	parent := len/2 - 1
	for parent >= 0 {
		heapify(nums, parent, len)
		parent--
	}
}

func heapify(nums []int, parent, len int) {
	// 判斷兩個子節(jié)點(diǎn)是否比父節(jié)點(diǎn)大，如果是的話替換
	max := parent
	lson := parent*2 + 1
	rson := parent*2 + 2
	if lson < len && nums[lson] > nums[max] {
		// 左節(jié)點(diǎn)是否大于父節(jié)點(diǎn)
		max = lson
	}
	if rson < len && nums[rson] > nums[max] {
		// 右節(jié)點(diǎn)是否大于父節(jié)點(diǎn)
		max = rson
	}
	if parent != max {
		swap(&nums[max], &nums[parent])
		heapify(nums, max, len)
	}
}
nums :=[]int{3, 5, 3, 0, 8, 6}
buildHeep(nums,len(nums))
// 結(jié)果 ： [8 5 6 0 3 3]

func HeapSort(nums []int) []int {
	// 堆排序,只能確認(rèn)第一次個數(shù)是最大或最小的
	// 調(diào)換第一個元素和最后一個元素位置、從0倒數(shù)第二個繼續(xù)堆排序
	i := len(nums)
	for i > 1 {
		buildHeep(nums, i)
		swap(&nums[0], &nums[i-1])
		i--
	}

	return nums
}

// heap.go
package structpk

import "fmt"

/*
	給定整數(shù)數(shù)組nums和k，
	請返回數(shù)組中第k個最大元素，
	請注意，你需要找的是數(shù)組排序后的第k個最大元素，
	而不是第k個不同的元素
*/
func swap(a, b *int) {
	*a, *b = *b, *a
}

func HeapSort(nums []int) []int {
	// 堆排序,只能確認(rèn)第一次個數(shù)是最大或最小的
	// 調(diào)換第一個元素和最后一個元素位置、從0倒數(shù)第二個繼續(xù)堆排序
	i := len(nums)
	for i > 1 {
		buildHeep(nums, i)
		swap(&nums[0], &nums[i-1])
		i--
	}

	return nums
}
func buildHeep(nums []int, len int) {
	// 找到最后一個節(jié)點(diǎn)的父節(jié)點(diǎn)
	parent := len/2 - 1
	for parent >= 0 {
		heapify(nums, parent, len)
		parent--
	}
	fmt.Println(nums[0:len])

}

func heapify(nums []int, parent, len int) {
	// 判斷兩個子節(jié)點(diǎn)是否比父節(jié)點(diǎn)大，如果是的話替換
	max := parent
	lson := parent*2 + 1
	rson := parent*2 + 2
	if lson < len && nums[lson] > nums[max] {
		// 左節(jié)點(diǎn)是否大于父節(jié)點(diǎn)
		max = lson
	}
	if rson < len && nums[rson] > nums[max] {
		// 右節(jié)點(diǎn)是否大于父節(jié)點(diǎn)
		max = rson
	}
	if parent != max {
		swap(&nums[max], &nums[parent])
		heapify(nums, max, len)
	}
}

// main.go:
package main

import (
	"demo/structpk"
	"fmt"
)
func main() {

	fmt.Println(structpk.HeapSort([]int{
		3, 5, 3, 0, 8, 6,
	}))
}