堆排序-golang语言实现

最新推荐文章于 2024-07-31 16:52:17 发布

触不可及<>

最新推荐文章于 2024-07-31 16:52:17 发布

阅读量722

点赞数

分类专栏：计算机文章标签：堆排序排序算法数据结构算法 go语言

本文链接：https://blog.csdn.net/qq_33898609/article/details/121573190

版权

计算机专栏收录该内容

43 篇文章 1 订阅

订阅专栏

题目链接：力扣912：排序数组

使用数组存储堆，节点之间的索引关系：
本文从数组索引0位置开始存储
若当前节点索引为i，则父节点为 (i - 1) / 2
两个子节点分别为： 2i + 1 ， 2i + 2

如下图所示：
在这里插入图片描述堆排序主要有两步：
建堆
排序（本文构建大顶堆，进行升序排序）

(1) 建堆：对于完全混乱的数组，从0开始建堆，只需要从最后一个叶子节点的父节点开始往前遍历，分别执行建堆操作，即可完成堆的构建。如下图所示：
在这里插入图片描述

(2) 排序：利用堆进行排序，每次操作将堆顶元素(保存的最大值)和最后一个元素进行交换，并对新的堆的元素进行下沉操作。遍历堆中所有元素，循环上面操作。
排序详细讲解见下方视频链接22:50

func sortArray(nums []int) []int {
    length := len(nums)

    // 下沉建堆,从最后一个结点的父节点开始，往前遍历，执行建堆操作
    for i := (length - 1) / 2; i >= 0; i-- {
        sink(nums, i, length)
    }

    // 排序
    for k := length - 1; k >= 0 ;k-- {        
        nums[0], nums[k] = nums[k], nums[0]
        sink(nums, 0, k)
    }
    return nums
}

func sink(nums []int, i, end int) {
    for 2 * i <= end {
        // 获取子节点
        c1 := 2 * i + 1
        c2 := 2 * i + 2
        max := i    //默认当前i索引所在结点为最大值
        // 寻找当前结点和两个子节点中的最大值
        if c1 < end && nums[i] < nums[c1] {
            max = c1
        }
        if c2 < end && nums[i] < nums[c2] {
            max = c2
        }
        // 交换操作，父节点下沉，与最大的孩子节点交换
        if max != i {   //如果最大值是孩子结点，则进行交换，否则建堆完成，退出
            nums[max], nums[i] = nums[i], nums[max]
        } else {
            break
        }
        // 继续下沉
        i = max
    }
}