您的位置:首页 > 编程语言 > Go语言

【golang必备算法】堆

2021-11-04 16:43 471 查看
堆:

一般我们提到的是二叉堆,是一种完全二叉树,二叉堆有两种:最大堆和最小堆。特点是父节点的值大于(小于)子节点


基础知识

完全二叉树有一个性质,除了最底层,每一层都是满的,这使得堆可以利用数组来表示,如下图

对给定某个结点下标i,可以计算出这个结点的父节点、孩子节点的下标:

父节点下标:i/2

左子节点下标:2i

右子节点下标:2i+1

但是数组都是0开始的,所以调整一下,如图:

父节点下标:(i - 1)/2

左子节点下标:2i + 1

右子节点下标:2(i + 1)


维护堆

维护堆的性质过程如下图

由于调整一次,堆任然有可能违反堆性质,所以需要递归的测试,使整个堆都满足堆性质


建堆

将一个数组改造成一个堆,如图将其变成一个大顶堆

数组长度为l,首先找到(l / 2)-1 处,也就是下标为4的节点然后开始维护堆

然后自下而上,维护每一个节点,直到根节点

自下而上建堆比自上而下建堆效率高
go 实现堆排序

package main

import "fmt"

func main(){

arr:=[]int{4,8,2,1,6,9,3,5,7,8,1,4}
dui(arr)
fmt.Println(arr)
}

func swap(arr []int,a,b int){
arr[a],arr[b]=arr[b],arr[a]
}

func heapadjust(arr []int,i,m int){

son:=i*2+1
for son<m{
if son+1<m&&arr[son]<arr[son+1]{//首先判断选出左节点大还是右节点大
son++
}
if arr[i]>arr[son]{
break
}else{
swap(arr,i,son)
i=son
son=i*2+1
}
}
}

func dui(arr []int){
for i:=len(arr)/2-1;i>=0;i--{	//这里是先变成大顶堆
heapadjust(arr,i,len(arr))
}

for i:=len(arr)-1;i > 0;i--{	//将最大值与最后一个交换,然后对前i-1重新排序
swap(arr,0,i)
heapadjust(arr,0,i-1)

}
}
内容来自用户分享和网络整理,不保证内容的准确性,如有侵权内容,可联系管理员处理 点击这里给我发消息
标签: