package main

import "fmt"

//堆的特性
//1.是完全二叉樹
//2.每一個節點都大於子節點(大堆)
//3.根節點最大
//4.左子節點2i+1, 由子節點2i+2, 父節點(i-1)/2
func parentNode(i int)  int{
	return (i - 1)/2
}

//左節點
func leftNode(i int) int{
	return 2*i + 1
}
//右節點
func rightNode(i int) int{
	return 2*i + 2
}
//建立heap
func buildHeap(heap []int) int{
	length := len(heap)
	for i := length/2 - 1; i >= 0; i-- {
		maxHeap(heap, i, length)
	}
	return length
}

func maxHeap(heap []int, i int, length int) {
	left := leftNode(i)
	right := rightNode(i)
	largest := 0
	if left < length && heap[left] > heap[i] {
		largest = left
	}else {
		largest = i
	}
	if right < length && heap[right] > heap[largest] {
		largest = right
	}
	if largest != i {
		heap[i], heap[largest] = heap[largest], heap[i]
		//需要繼續比較起父節點
		maxHeap(heap, largest, length)
	}
}

func main() {
	a := []int{3,31,3,54,5,6,7,34,2,213}
	heapLength := buildHeap(a)
	for i := len(a) - 1; i > 0; i-- {
		a[i], a[0] = a[0], a[i] // 把第一個數字 是最大的 和最後一個交換 最後一個就 i不斷減少 最後的 倒數第二 倒數第三 不斷是當前最大的數字
		heapLength--
		maxHeap(a, 0, heapLength)
	}
	fmt.Println(a)
}