温馨提示×

golang堆排序怎么实现

小亿
103
2023-10-26 16:27:56
栏目: 编程语言

Golang堆排序的实现步骤如下:

  1. 首先,创建一个函数heapify用于将给定的数组或切片转换为一个最大堆。最大堆的定义是父节点的值大于或等于其子节点的值。
func heapify(arr []int, n int, i int) {
	largest := i  // 将当前节点标记为最大值
	left := 2*i + 1  // 左子节点的索引
	right := 2*i + 2  // 右子节点的索引

	// 如果左子节点存在且大于根节点,则将最大节点的索引更新为左子节点
	if left < n && arr[left] > arr[largest] {
		largest = left
	}

	// 如果右子节点存在且大于根节点,则将最大节点的索引更新为右子节点
	if right < n && arr[right] > arr[largest] {
		largest = right
	}

	// 如果最大节点的索引不是根节点,则将最大节点和根节点交换,并递归地对交换后的子树进行堆化
	if largest != i {
		arr[i], arr[largest] = arr[largest], arr[i]
		heapify(arr, n, largest)
	}
}
  1. 接下来,创建一个函数heapSort来执行堆排序。
func heapSort(arr []int) {
	n := len(arr)

	// 构建最大堆(初始化堆)
	for i := n/2 - 1; i >= 0; i-- {
		heapify(arr, n, i)
	}

	// 逐个提取堆中的元素,并进行堆排序
	for i := n - 1; i >= 0; i-- {
		// 将当前根节点(最大值)与数组的最后一个元素交换
		arr[0], arr[i] = arr[i], arr[0]

		// 重新构建最大堆,但不包括已经排序的元素
		heapify(arr, i, 0)
	}
}
  1. 最后,调用heapSort函数来对给定的数组进行堆排序。
func main() {
	arr := []int{12, 11, 13, 5, 6, 7}
	heapSort(arr)
	fmt.Println("排序后的数组:", arr)
}

这样就完成了使用Golang实现堆排序的过程。输出结果将会是[5 6 7 11 12 13]

0