go 堆排序

最新推荐文章于 2024-06-15 23:36:03 发布

bob62856

最新推荐文章于 2024-06-15 23:36:03 发布

阅读量50

点赞数

分类专栏： Algorithm 文章标签：算法数据结构贪心算法

本文链接：https://blog.csdn.net/qq_20817327/article/details/130656737

版权

Algorithm 专栏收录该内容

374 篇文章 3 订阅

订阅专栏

func heapSort(arr []int, start, end int) {
    dad := start
    left_son := 2*dad + 1
    right_son := 2*dad + 2
    if left_son > end {
        return
    }
    idx := left_son
    if right_son <= end {
        if arr[right_son] > arr[left_son] {
            idx = right_son
        }
    }
    if arr[dad] > arr[idx] {
        return
    }
    swap(arr, dad, idx)
    heapSort(arr, idx, end)
}

func main(arr []int) []int {
    n := len(arr)
    for i := n/2 - 1; i >= 0; i-- {
        heapSort(arr, i, n-1)
    }

    for i := n - 1; i > 0; i-- {
        swap(arr, 0, i)
        heapSort(arr, 0, i-1)
    }

    return arr[]
}

func swap(arr []int, a, b int) {
    tmp := arr[a]
    arr[a] = arr[b]
    arr[b] = tmp
}