堆的实现
一、关于堆的一些基本认识
1、什么是堆?
如果有一个关键码的集合K = { , , ,…, },把它的所有元素按完全二叉树的顺序存储方式存储
在一个一维数组中,并满足: <= 且 <= ( >= 且 >= ) i = 0,1,2…,则称为小堆(或大堆)。将根节点最大的堆叫做最大堆或大根堆,根节点最小的堆叫做最小堆或小根堆。
大根堆:孩子的值小于等于父节点的值;
小根堆:孩子的值大于等于父节点的值;
2、堆的性质:
- 堆中某个节点的值总是不大于或不小于其父节点的值;
- 堆总是一棵完全二叉树。
3、 什么是完全二叉树?
满二叉树:一个二叉树,如果每一个层的结点数都达到最大值,则这个二叉树就是满二叉树。也就是说,如果一个二叉树的层数为K,且结点总数是 ,则它就是满二叉树。
完全二叉树:完全二叉树是效率很高的数据结构,完全二叉树是由满二叉树而引出来的。对于深度为K的,有n个结点的二叉树,当且仅当其每一个结点都与深度为K的满二叉树中编号从1至n的结点一一对应时称之为完全二叉树。 要注意的是满二叉树是一种特殊的完全二叉树。就是前K-1层都是满的,最后一层不是满的,最后一层节点数1~2^(k-1),就是最后一层只有一个叶子结点到最后一层是满的。
4、实现对三个关于父子节点关系的公式:
因为在内存里是以数组形式存储的,为方便理解,如下图,数组坐标在逻辑上的位置
注意:要把它看做根,根节点的左子树,根节点的右子树。再把左子树看做一个树,又是根,左子树右子树…
二、堆的插入
在堆的后面插入一个新的数,让它仍然保持小根堆(或大根堆)
- 插入前需要判断容量,容量不够需要扩容realloc,如果为空,给初始几个空间。
- 容量方面OK,将数放入数组。
- 然后通过向上调成算法使它维持大根堆或者小根堆。
1、向上调整算法
思路:如果是调整成大根堆,从倒数第一个非叶子结点开始(其实就是倒数第1行的叶子结点的父亲),如果,它比它的的左右孩子小,把左右孩子较大的那一个孩子和父亲节点交换值,父结点成为新的孩子求出自己的父节点的值进行比较。如果父亲比孩子大,那么不用动。
- 父节点下标=(孩子下标-1)/2
- 孩子的下标要等于0时停止,因为到了根节点
如图;在大根堆的后面插入100,让它仍然保持大根堆
同理向上调整成小根堆是类似的
思路:如果是调整成小根堆,从倒数第一个非叶子结点开始(其实就是倒数第1行的叶子结点的父亲),如果,它比它的的左右孩子大,把左右孩子较小的那一个孩子和父亲节点交换值,父结点成为新的孩子求出自己的父节点的值进行比较。如果父亲比孩子小,那么不用动。 - 父节点下标=(孩子下标-1)/2
- 孩子的下标要等于0时停止,因为到了根节点
如图;在小根堆的后面插入3,让它仍然保持小根堆
//传数组地址和孩子下标
void AdjustUp(HeapDataType* a, int child)
{
int parent = (child - 1) / 2;
//到零已经是根节点了
while (child > 0)
{
if (a[parent] > a[child])
{
/*int tmp = a[parent];
a[parent] = a[child];
a[child] = tmp;*/
//可用swap函数代替,尽量用函数
Swap(&a[parent], &a[child]);
child = parent;//更新孩子下标
parent = (child - 1) / 2;//更新父亲下标
}
else
break;
}
}
//插入
//首先判断空间是不是足够,如果当size=capactiy时说明满了
//为空的话给一定的空间
void HeapPush(Heap* hp, HeapDataType x)
{
assert(hp);
//当capacity==size有为空或者满了的情况,需要判断
if (hp->capacity == hp->size)
{
int newcapacity =hp->capacity== 0 ? hp->capacity = 4 : hp->capacity * 2;
HeapDataType* tmp = (HeapDataType*)realloc(hp->a, sizeof(HeapDataType) * newcapacity);
if (tmp == NULL)
{
perror("relloc fair");