堆的实现
一、关于堆的一些基本认识
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");
exit(-1);
}
hp->a = tmp;//把新地址传给a
hp->capacity = newcapacity;
}
hp->a[hp->size] = x;//因为是数组下标所以是size
hp->size++;//个数加一
//建小堆,孩子比父节点小;建大堆,孩子比父节点大
//进行向上调整
//比较孩子与父亲的大小,有孩子的下标可以知道父亲的
//parent=(child-1)/2 leftchild=parent*2+1 rightchild=parent*2+2
AdjustUp(hp->a, hp->size -1);
}
三、删除堆顶元素(找次大或者次小)
将根节点结点的值与最后一个叶子结点值进行交换,然后删除最后一个叶子结点(数组size-1,就是删除最后一个叶子结点),将前n-1个结点作为新的树,然后进行向下调整,使堆保持大根堆或者小根堆。
1.交换第一个和最后一个
2.向下调整
1、 向下调整算法
思路:如果是调整成大根堆,从根节点开始,如果父亲比孩子小,父节点的值与左右孩子较大的值进行交换,然后较大的孩子的位置成为新的父节点,继续与自己的孩子进行比较。
- 先假设左孩子比右孩子大,如果假设错了那么较大的就是右孩子
- 孩子的下标要<n,到最后一个节点的时候就结束了
同理向下调整成小根堆是类似的
思路:如果是调整成小根堆,从根节点开始,如果父亲比孩子大,父节点的值与左右孩子较小的值进行交换,然后较小的孩子的位置成为新的父节点,继续与自己的孩子进行比较。 - leftchild=parent*2+1(数组下标)
- rightchild=parent*2+2(数组下标)
- 先假设左孩子比右孩子小,如果假设错了那么较小的就是右孩子
- 孩子的下标要<n,到最后一个节点的时候就结束了
//向下调整
void AdjustDown(HeapDataType* a, int size,int parent)
{
//假设左孩子比右孩子小
int minchild = parent * 2 + 1;
//size是元素个数,child的下标不超过size-1就好,超过了,那个下标么有孩子
while (minchild<=size-1)
{
//如果右孩子比左孩子小,那右孩子是minchild
if (minchild + 1 <size - 1 && a[minchild] > a[minchild + 1])
{
minchild++;
}
//如果父亲比孩子大,交换值,换到下面
if (a[parent] > a[minchild])
{
Swap(&a[parent], &a[minchild]);
//更新
parent = minchild;
//假设左孩子比右孩子小
minchild = parent * 2 + 1;
}
//如果比左右孩子都小,就不用动了
else
break;
}
}
删除堆顶元素
void HeapPop(Heap* hp)
{
assert(hp);
assert(!isEmpty(hp));
Swap(&hp->a[0], &hp->a[hp->size - 1]);
hp->size--;
AdjustDown(hp->a, hp->size-1,0);
}
四、堆的创建
1、 利用向上调整算法建堆
模拟插入过程:
时间复杂度N x logN
for(i=1;i<n;i++)
{
Adjustup(a,i);
}
2、将已有的数组建堆
给出一个数组int a[ ]={1,5,3,8,7,6};
这个数组可以在逻辑上看做一个完全二叉树,但他还不是一个堆,可以通过向上调整算法或者向下调整算法,将它调整成一个堆。
利用向上调整算法建堆
- 从最后一个叶子节点的父节点开始开始,如果根节点的值比孩子小,左右孩子比较,把两者大的孩子换上去,根节点与那个大的孩子交换位置。
最后一个叶子下标n-1,它的父节点parent=(child-1)/2=(n-1-1)/2
for(i=((n-1)-1)/2;i>0;i--)
{
AdjustDown(a,n,i);
}
将数组利用向上调整算法构建成大根堆
代码
** heap.h**
#pragma once
#include <stdio.h>
#include <stdlib.h>
#include <assert.h>
#include <stdbool.h>
typedef int HeapDataType;
typedef struct Heap
{
HeapDataType* a;
int size;
int capacity;
}Heap;
//初始化
void HeapInit(Heap* hp);
//打印
void HeapPrint(Heap* hp);
//插入
void HeapPush(Heap* hp, HeapDataType x);
//向上调整
void AdjustUp(HeapDataType* a, int child);
//删除堆顶元素
void HeapPop(Heap* hp);
//向下调整
void AdjustDown(HeapDataType* a, int parent);
//释放数组
void DestroyHeap(Heap* hp);
//返回栈顶元素
HeapDataType HeapTop(Heap* hp);
//判空
bool isEmpty(Heap* hp);
int HeapSize(Heap* php);
heap.c
#include "Heap.h"
//初始化
void HeapInit(Heap* hp)
{
assert(hp);
hp->a = NULL;
hp->capacity = hp->size = 0;
}
//打印
void HeapPrint(Heap* hp)
{
assert(hp);
for (int i = 0; i < hp->size; i++)
{
printf("%d ", hp->a[i]);
}
printf("\n");
}
void Swap(HeapDataType* parent, HeapDataType* child)
{
int tmp = *parent;
*parent = *child;
*child= tmp;
}
//传数组地址和孩子下标
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");
exit(-1);
}
hp->a = tmp;//把新地址传给a
hp->capacity = newcapacity;
}
hp->a[hp->size] = x;//因为是数组下标所以是size
hp->size++;//个数加一
//建小堆,孩子比父节点小;建大堆,孩子比父节点大
//进行向上调整
//比较孩子与父亲的大小,有孩子的下标可以知道父亲的
//parent=(child-1)/2 leftchild=parent*2+1 rightchild=parent*2+2
AdjustUp(hp->a, hp->size -1);
}
//向下调整
void AdjustDown(HeapDataType* a, int size,int parent)
{
//假设左孩子比右孩子小
int minchild = parent * 2 + 1;
//size是元素个数,child的下标不超过size-1就好,超过了,那个下标么有孩子
while (minchild<=size-1)
{
//如果右孩子比左孩子小,那右孩子是minchild
if (minchild + 1 <size - 1 && a[minchild] > a[minchild + 1])
{
minchild++;
}
//如果父亲比孩子大,交换值,换到下面
if (a[parent] > a[minchild])
{
Swap(&a[parent], &a[minchild]);
//更新
parent = minchild;
//假设左孩子比右孩子小
minchild = parent * 2 + 1;
}
//如果比左右孩子都小,就不用动了
else
break;
}
}
bool isEmpty(Heap* hp)
{
assert(hp);
return hp->size == 0;
}
删除堆顶元素
void HeapPop(Heap* hp)
{
assert(hp);
assert(!isEmpty(hp));
Swap(&hp->a[0], &hp->a[hp->size - 1]);
hp->size--;
AdjustDown(hp->a, hp->size-1,0);
}
//释放数组
void DestroyHeap(Heap* hp)
{
free(hp->a);
hp->a = NULL;
hp->capacity = hp->size = 0;
}
//返回栈顶元素
HeapDataType HeapTop(Heap* hp)
{
assert(hp);
assert(!isEmpty(hp));
return hp->a[0];
}
int HeapSize(Heap* php)
{
assert(php);
return php->size;
}