定义
堆 :一种特殊的完全二叉树,若堆的根结点小于左儿子和右儿子,且他的左子树和右子树同样满足这一条件则称作小堆
若堆的根结点大于左儿子和右儿子,且他的左子树和右子树同样满足这一条件则称作大堆
二叉树的顺序存储
堆中的元素以顺序存储的方式存储,顺序结构存储就是使用数组来存储,一般使用数组只适合表示完全二叉树,因为不是完全二叉树会有空间的浪费。
完全二叉树顺序存储如上图所示,下标0到8表示不同结点
观察结点的下标 可以发现:
1.左孩子 = 父亲结点X2+1
2.右孩子 = 左孩子 +1
3.父亲结点 = (左/右孩子 -1 )/2
堆的创建和初始化
先创建一个堆
typedef struct Heap//堆
{
DataType* array;
int capacity;//记录堆的大小
int size;//堆中元素个数
}Heap;
初始化
void InitHeap(Heap*h,int size)//堆初始化
{
h->array = (DataType*)malloc(sizeof(DataType)*(size));
assert(h->array);
h->capacity = size;
h->size = 0;
}
向下调整
先以下为例:
一棵树左子树为堆 右子树也是堆 怎样调整根结点 是整个树成为堆
void AdjustDown(int *array, int size, int parent)
{//size 表示结点个数 parent表示父结点位置 也就是需要调整的结点的位置
//时间复杂度为log2 n 即二叉树的深度
int child = parent * 2 + 1;//可能没有右孩子
while (child < size)
{
if (child + 1 < size&&array[child + 1] < array[child])//让child等于左右孩子中小的哪一个
{
child += 1;
}
if (array[child] < array[parent])//如果parent比左右儿子大
{
swap(&array[child], &array[parent]);
parent = child;//交换一次后 可能不再是堆 继续判断下去
child = parent * 2 + 1;
}
else
return;
}
}
创建
从二叉树的最后一个叶子结点的根开始往上创建 直到整棵树的根结点
last 表示最后一个叶子结点的根 last–表示倒数第二个叶子结点的根 。。。
void CreatHeap(Heap*h, int*array, int size)//复杂度为nlogn
{
//找倒数第一个非叶子结点
InitHeap(h,size);
int last = (size - 1-1) / 2;
for (int i = 0; i < size; i++)
{
h->array[i] = array[i];
h->size++;
}
for (last; last >= 0; last--)//从最后一个根结点往后进行 复杂度为n/2
{
AdjustDown(h->array, h->size, last);
}
}
插入元素
将新的元素放在末尾 向上调整 直到根结点
需要开辟新的空间
void Insert(Heap*h, DataType data)
{
assert(h != NULL);
h->array = (DataType*)realloc(h->array, sizeof(DataType)*(h->size + 1));
h->array[h->size] = data;
h->size++;
h->capacity++;
int child = h->size - 1;
int parent = (child - 1) / 2;
while (parent > 0)
{
if (h->array[parent] > h->array[child])
{
swap(&h->array[parent], &h->array[child]);
child = parent;
parent = (parent - 1) / 2;
}
if (parent== 0&&h->array[parent] > h->array[child])
{
swap(&h->array[parent], &h->array[child]);
}
}
}
删除堆顶元素
删除堆顶元素后 要对堆进行调整 使其从新变成一个堆
方法:
将最后一个结点 放到堆顶 然后向下调整
void EraseHeap(Heap*h)//删除堆顶元素
{//堆顶元素和最后一个元素交换 然后在 Adjustdown
assert(h);
if (h->size == 0)
return;
swap(&h->array[0], &h->array[h->size - 1]);
h->size--;
AdjustDown(h->array, h->size, 0);//0表示parent位置
}