go语言堆排序怎么实现

   2025-02-15 3060
核心提示:Go语言堆排序的实现步骤如下:首先,定义一个用于进行堆调整的函数 adjustHeap,该函数接受三个参数:待调整的切片 arr,当前需

Go语言堆排序的实现步骤如下:

首先,定义一个用于进行堆调整的函数 adjustHeap,该函数接受三个参数:待调整的切片 arr,当前需要调整的节点的下标 i,以及堆的大小 length。在 adjustHeap 函数中,首先获取当前节点的值,然后计算出其左子节点和右子节点的下标。比较左子节点和右子节点的值,取较大值的下标作为 maxIndex。判断当前节点与其子节点的大小关系,如果当前节点的值小于 maxIndex 所对应的子节点的值,则交换两者的值,并递归调用 adjustHeap 函数,以保证堆的性质。在主函数中,首先构建一个初始堆,通过调用 adjustHeap 函数来从最后一个非叶子节点开始进行堆调整。将堆顶元素与最后一个元素交换,然后将堆的大小减一,并调用 adjustHeap 函数对堆顶元素进行调整,以保持堆的性质。重复步骤 6,直到堆的大小为 1,此时,整个序列已经有序。

下面是具体的代码实现:

package mainimport "fmt"func adjustHeap(arr []int, i, length int) {temp := arr[i]for k := i*2 + 1; k < length; k = k*2 + 1 {if k+1 < length && arr[k] < arr[k+1] {k++}if arr[k] > temp {arr[i] = arr[k]i = k} else {break}}arr[i] = temp}func heapSort(arr []int) {length := len(arr)for i := length/2 - 1; i >= 0; i-- {adjustHeap(arr, i, length)}for i := length - 1; i > 0; i-- {arr[0], arr[i] = arr[i], arr[0]adjustHeap(arr, 0, i)}}func main() {arr := []int{9, 8, 7, 6, 5, 4, 3, 2, 1}heapSort(arr)fmt.Println(arr)}

输出结果为 [1 2 3 4 5 6 7 8 9],表示已经成功对输入的序列进行了堆排序。

 
 
更多>同类维修知识
推荐图文
推荐维修知识
点击排行
网站首页  |  关于我们  |  联系方式  |  用户协议  |  隐私政策  |  网站留言