堆排序(GoLang)

heapInsert函数:将加入的节点的数与父节点的数值进行比较,若父节点的数值大于该新加入节点的数值,那么就不做变化,若父节点的数值小于该新加入节点的数值,那么就将该节点与父节点进行调换,然后继续再与下一个父节点进行比较,直到该节点成为根节点或者父节点大于该节点的情况就停。这样做的目的是为了将此完全二叉树变成一个大根堆。

// index : insert的值在arr中的索引
// 将arr 变成大根堆
func heapInsert(arr *[]int, index int) {
	// 父节点为 (index - 1)/2
	for (*arr)[(index-1)/2] < (*arr)[index] {
		//fmt.Println("heapInsert")
		moddle.Swap(arr, (index-1)/2, index)
		index = (index - 1) / 2
	}

heapIfy函数:根节点(或任意一节点)更换了数值,则比较其与其左右孩子中较大的那个孩子的大小,若小于则交换位置,直到遇到不小于左右孩子或者到了heapSize的位置(heapSize是该大根堆的节点总数)。 

func heapIfy(arr *[]int, index int, heapSize int) {
	// 首先要判断根节点是否还存在左孩子,如果存在左孩子,说明还有向下比较大小,直到变成大根堆
	left := index*2 + 1
	// 只要left小于heapSize,就说明index至少还存在一个孩子
	for left < heapSize {
		//fmt.Println("heapIfy", left, " ", heapSize, " ", index)
		//moddle.PrintArr(*arr)
		var largest int
		// 判断左右孩子是否存在和比较左右孩子大小
		if left+1 < heapSize && (*arr)[left+1] > (*arr)[left] {
			largest = left + 1
		} else {
			largest = left
		}
		// 判断孩子最大值是否比index大
		if (*arr)[largest] < (*arr)[index] {
			break
		}
		moddle.Swap(arr, index, largest)
		index = largest
		left = index*2 + 1
	}
}

heapSort函数:将数组做heapInsert,变成大根堆,然后将根节点与最后一个节点交换位置,此时数组中的一个值已经来到了正确的位置,heapSize--(代表大根堆中需要排序的数值减少了一个),然后做heapIfy(因为此时不是大根堆,做heapIfy将其再变成大根堆),然后再重复以上操作。

func HeapSort(arr *[]int) {
	n := len(*arr)
	for i := 0; i < n; i++ {
		heapInsert(arr, i)
	}
	//moddle.PrintArr(*arr)
	heapSize := n - 1
	// 将大根堆的根节点(也就是数组的第一个数值与最后一个数交换位置)放到最后,这样就确定好了第一个最大的数
	moddle.Swap(arr, 0, heapSize)
	for heapSize > 0 {
		heapIfy(arr, 0, heapSize)
		heapSize--
		moddle.Swap(arr, 0, heapSize)
	}

}

时间复杂度O(NlogN)

空间复杂度O(1)

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值