堆排序(heapSort)

堆可以分为大根堆和小根堆,是一个完全二叉树
堆排序是根据堆结构设计的一种排序

堆的分类
大根堆:每个结点的值都大于其左孩子和右孩子结点的值
小根堆:每个结点的值都小于其左孩子和右孩子结点的值

数组中某元素的父结点和左右孩子结点,比如已知索引为i的数,那么
1)父结点索引:(i-1)/2
2)左孩子索引:2i+1
3)右孩子索引:2
i+2

堆排序算法思路:
1.首先将数组构造成一个大根堆,堆结构的顶端值为整个数组的最大值
2.将顶端的数与末尾的数交换,此时,末尾的数为最大值,剩余待排序数组个数为 size - 1
3.将剩余的 size - 1 个数再构造成大根堆,再将顶端数与 size - 1 位置的数交换,如此反复执行,便能得到有序数组

平均时间复杂度:O(nlogn)
初始化建堆:O(n)
排序重建堆:O(nlog(n))
总复杂度:O(n+nlogn)=O(nlogn)
空间复杂度:O(1)
稳定性不稳定,非堆顶元素在向堆顶移动的过程中,经历着堆层次的改变,这就有可能导致相等元素相对位置的改变
适合场景:待排序序列中元素个数较多,堆排序效率较高。

package main

import "fmt"

func main() {
	arr := []int{1, 3, 6, 5, 2, 8}
	heapSort(arr)
	for _, v := range arr {
		fmt.Println(v)
	}
}

// 堆排序:
func heapSort(arr []int) {
	// 所有元素入堆
	/*// 自上而下建堆:O(nlogn)
	for i := 0; i < len(arr); i++ {
		heapInsert(arr, i)
	}*/
	// 自下而上建堆:O(n)
	for i := len(arr) - 1; i >= 0; i-- {
		heapify(arr, i, len(arr))
	}
	// 将首元素(最大值元素)弹出(与尾值交换)
	size := len(arr)
	arr[0], arr[size-1] = arr[size-1], arr[0]
	size--
	// 不断弹出最大元素,完成排序:
	for size > 0 {
		heapify(arr, 0, size)
		arr[0], arr[size-1] = arr[size-1], arr[0]
		size--
	}
}

// 新元素插入堆操作:向上操作
func heapInsert(arr []int, index int) {
	for arr[index] > arr[(index-1)/2] {
		arr[index], arr[(index-1)/2] = arr[(index-1)/2], arr[index]
		index = (index - 1) / 2
	}
}

// 剩余元素重新成堆:向下操作
func heapify(arr []int, index, size int) {
	left := 2*index + 1
	for left < size {
		right := left + 1
		largest := left
		if right < size && arr[right] > arr[left] {
			largest = right
		}
		if arr[index] > arr[largest] {
			largest = index
		}
		if largest == index {
			break
		}
		arr[largest], arr[index] = arr[index], arr[largest]
		index = largest
		left = 2*index + 1
	}
}
  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

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

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

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值