目录
- 一、树的相关概念及其特殊二叉树
- 1、数的相关概念
- 2、特殊二叉树
- 二、堆
- 1、堆的实现
- 1.1、堆向下调整算法和向上调整算法的时间复杂度
- 1.2、堆的构建
- 1.3、堆的插入
- 1.4、堆的删除
- 1.5、取堆顶的数据、堆的个数及堆的判空
- 2、堆的排序
一、树的相关概念及其特殊二叉树
讲堆之前，我们先讲讲树的相关概念及其特殊二叉树，因为堆是一种二叉树，是一棵完全二叉树。
1、数的相关概念
- 节点的度：一个节点含有的子树的个数，如A的度为3
- 叶子节点：度为0的节点，如F、L、H、M等
- 非叶子节点：度不为0的节点，如B、C、D、G等
- 双亲节点或父节点：若一个节点有子节点，则这个节点称为其子节点的父节点，如A是B的父节点
- 孩子节点或子节点：一个节点含有子树的根节点称为该节点的子节点，如B是A的子节点
- 兄弟节点：具有树同父节点的节点互称为兄弟节点，如B、C是兄弟节点
- 树的度：一棵树中，节点最大的度是树的度，如上图树的度为3
- 节点的层次：从根开始定义起，根为第一层，根的子节点为第二层，依次类推
- 树的高度或深度：树中节点的最大层次为树的高度，如上图树的高度为4
- 堂兄弟节点：双亲在同一层的节点互为堂兄弟，如L、M互为堂兄弟节点
- 节点的祖先：从根到该节点所经分支上的所以节点，如A是所以节点的祖先
- 子孙：以某节点为根的子树中任一节点都称为该节点的子孙，如所以节点都是A的子孙
- 森林：由m(m>0)棵互不相交的树的集合称为森林
2、特殊二叉树
二叉树是由一个根节点加上左子树和右子树组成
2.1、满二叉树
满二叉树每一个层的节点数都达到最大值
若一个满二叉树有K层，则第K层的节点数为2^(K-1)
总节点数为2^K-1
假设一个满二叉树有N个节点，则该树的高度为h=log(N+1)
2.2、完全二叉树
完全二叉树前N-1层是满的，最后一层可以不满，但是必须是从左到右是连续的
假设完全二叉树的高度是h
最多节点数为2^h-1
最少节点数为2^(h-1)
对任何一棵二叉树，如果度为0的叶子节点个数为N0，度为2的分支节点个数为N2，则有N0=N2+1，即度为0的节点总比度为2的节点多一个
二、堆
1、堆的实现
堆分为大堆和小堆，实际存储在一个数组当中。
大堆：树中所有父亲都大于等于孩子
小堆：树中所有父亲都小于等于孩子
堆的性质：
- 堆中某个节点的值总是不大于或不小于其父节点的值
- 堆总是一棵完全二叉树
堆孩子和父亲的下标关系
parent = (child - 1) / 2
leftchild = parent * 2 +1
rightchild = parent * 2 +2
1.1、堆向下调整算法和向上调整算法的时间复杂度
堆调整算法有一个要求：左右子树都必须是一个堆，才能调整
1、向下调整
以满二叉树为例：
由此可知，向下调整的时间复杂度为O(N)
代码实现：
void AdjustDown(HPDataType* a, int n, int parent)
{
int child = parent * 2 + 1;
while (child < n)
{
// 确认child指向大的那个孩子
if (child + 1 < n && a[child + 1] > a[child])
{
++child;
}
// 1、孩子大于父亲，交换，继续向下调整
// 2、孩子小于父亲，则调整结束
if (a[child] > a[parent])
{
Swap(&a[child], &a[parent]);
parent = child;
child = parent * 2 + 1;
}
else
{
break;
}
}
}
以上代码建大堆，若要建小堆，则把a[child + 1] > a[child] 和 if (a[child] > a[parent]) 改为 a[child + 1] < a[child] 和 if (a[child] < a[parent])
2、向上调整
由此可知，向下调整的时间复杂度为O(N*logN)
代码实现：
void AdjustUp(HPDataType* a, int child)
{
int parent = (child - 1) / 2;
while (child > 0)
{
if (a[child] > a[parent])
{
Swap(&a[child], &a[parent]);
child = parent;
parent = (child - 1) / 2;
}
else
{
break;
}
}
}
以上代码建大堆，若要建小堆，则把if (a[child] > a[parent]) 改为 if (a[child] < a[parent])
总结：
向下调整：节点多，调整少，节点少，调整多
向上调整：节点少，调整多，节点多，调整多
所以建堆，建议用向下调整
1.2、堆的构建
我们给一个数组，但还不是一个堆，我们可以通过建堆算法，把它构成一个堆。但是根节点的子树都不是堆，那我们怎么调整呢？我们可以从倒数的第一个非叶子节点的子树开始调整，一直调整到根节点的树，就形成堆了。
此时就构建出一个大堆：
代码实现：
#include <stdio.h>
#include <assert.h>
#include <stdlib.h>
#include <string.h>
typedef int HPDataType;
typedef struct Heap
{
HPDataType* a;
int size;
int capacity;
}HP;
extern void AdjustDown(HPDataType* a, int n, int parent);
// 堆的构建
void HeapCreate(HP* php, HPDataType* a, int n)
{
assert(php);
php->a = (HPDataType*)malloc(sizeof(HPDataType) * n);
if (php->a == NULL)
{
perror("malloc fail");
exit(-1);
}
memcpy(php->a, a, sizeof(HPDataType) * n);
php->size = php->capacity = n;
//建堆算法
for (int i = (n - 1 - 1) / 2; i >= 0; i--)
{
AdjustDown(php->a, n, i);
}
}
// 初始化堆
void HeapInit(HP* php)
{
assert(php);
php->a = NULL;
php->size = php->capacity = 0;
}
// 销毁堆
void HeapDestroy(HP* php)
{
assert(php);
free(php->a);
php->a = NULL;
php->size = php->capacity = 0;
}
// 交换
void Swap(HPDataType* p1, HPDataType* p2)
{
HPDataType tmp = *p1;
*p1 = *p2;
*p2 = tmp;
}
// 打印堆
void HeapPrint(HP* php)
{
assert(php);
for (int i = 0; i < php->size; ++i)
{
printf("%d ", php->a[i]);
}
printf("\n");
}
// 向下调整
void AdjustDown(HPDataType* a, int n, int parent)
{
int child = parent * 2 + 1;
while (child < n)
{
// 确认child指向大的那个孩子
if (child + 1 < n && a[child + 1] > a[child])
{
++child;
}
// 1、孩子大于父亲，交换，继续向下调整
// 2、孩子小于父亲，则调整结束
if (a[child] > a[parent])
{
Swap(&a[child], &a[parent]);
parent = child;
child = parent * 2 + 1;
}
else
{
break;
}
}
}
int main()
{
HP hp;
int a[] = { 1,5,3,8,7,6 };
HeapInit(&hp);
int n=sizeof(a) / sizeof(int);
HeapCreate(&hp, a, n);
HeapPrint(&hp);
HeapDestroy(&hp);
return 0;
}
结果：
1.3、堆的插入
先插入一个10到堆尾，再进行向上调整算法，直到形成堆
代码实现：
void HeapPush(HP* php, HPDataType x)
{
assert(php);
// 扩容
if (php->size == php->capacity)
{
int newCapacity = php->capacity == 0 ? 4 : php->capacity * 2;
HPDataType* tmp = (HPDataType*)realloc(php->a, sizeof(HPDataType) * newCapacity);
if (tmp == NULL)
{
perror("realloc fail");
exit(-1);
}
php->a = tmp;
php->capacity = newCapacity;
}
php->a[php->size] = x;
php->size++;
// 向上调整
AdjustUp(php->a, php->size - 1);
}
1.4、堆的删除
删除堆是删除堆顶元素，将堆顶的数据和最后一个数据交换，然后删除最后一个数据，在进行向下调整算法。
代码实现：
void HeapPop(HP* php)
{
assert(php);
assert(php->size > 0);
Swap(&php->a[0], &php->a[php->size - 1]);
php->size--;
AdjustDown(php->a, php->size, 0);
}
1.5、取堆顶的数据、堆的个数及堆的判空
代码如下：
//取堆顶的数据
HPDataType HeapTop(HP* php)
{
assert(php);
assert(php->size > 0);
return php->a[0];
}
//堆的个数
int HeapSize(HP* php)
{
assert(php);
return php->size;
}
// 堆的判空
bool HeapEmpty(HP* php)
{
assert(php);
return php->size == 0;
}
2、堆的排序
堆的排序分两个步骤：
1、建堆
- 升序：建大堆
- 降序：建小堆
2、利用堆删除思想来进行排序
- 把堆顶数据和最后一个数据进行交换，把最后一个数不看做堆里面的，相当于n-1个数，向下调整，选出次大的数。
代码如下：
#include <stdio.h>
typedef int HPDataType;
void Swap(HPDataType* p1, HPDataType* p2)
{
HPDataType tmp = *p1;
*p1 = *p2;
*p2 = tmp;
}
// 向下调整
void AdjustDown(HPDataType* a, int n, int parent)
{
int child = parent * 2 + 1;
while (child < n)
{
// 确认child指向大的那个孩子
if (child + 1 < n && a[child + 1] > a[child])
{
++child;
}
// 1、孩子大于父亲，交换，继续向下调整
// 2、孩子小于父亲，则调整结束
if (a[child] > a[parent])
{
Swap(&a[child], &a[parent]);
parent = child;
child = parent * 2 + 1;
}
else
{
break;
}
}
}
int main()
{
int a[] = { 1,5,3,8,7,6 };
int n = sizeof(a) / sizeof(int);
for (int i = (n - 1 - 1) / 2; i >= 0; --i)
{
AdjustDown(a, n, i);
}
int end = n - 1;
while (end > 0)
{
Swap(&a[0], &a[end]);
AdjustDown(a, end, 0);
--end;
}
for (int i = 0; i < n; i++)
{
printf("%d ", a[i]);
}
return 0;
}
运行结果：
TOP-K问题：即求数据结合中前K个最大的元素或者最小的元素
如：专业前10、世界500强等
举例：N个数中找最大的前K个数
第一种方法：建立一个N个数的大堆，删除K次，依次取堆顶
但是这个方法数据如果太大，就会放不进内存，直接存放在磁盘文件中，但是磁盘文件不能创建堆，所以次方法不适合。其时间复杂度为：O(N+logN*K)，空间复杂度为：O(1)
第二种方法：建立K个数的小堆，然后依次遍历数据，不堆顶大的数据，就替换堆顶，在向下调整，最后最大的K个数就在这个小堆里面。 时间复杂度：O(N*logK)，空间复杂度：O(K)
假设有一个data.txt文件中有以下数据：
要找出前5个最大的数，代码如下：
#include <stdio.h>
typedef int HPDataType;
void Swap(HPDataType* p1, HPDataType* p2)
{
HPDataType tmp = *p1;
*p1 = *p2;
*p2 = tmp;
}
// 向下调整
void AdjustDown(HPDataType* a, int n, int parent)
{
int child = parent * 2 + 1;
while (child < n)
{
// 确认child指向大的那个孩子
if (child + 1 < n && a[child + 1] < a[child])
{
++child;
}
// 1、孩子大于父亲，交换，继续向下调整
// 2、孩子小于父亲，则调整结束
if (a[child] < a[parent])
{
Swap(&a[child], &a[parent]);
parent = child;
child = parent * 2 + 1;
}
else
{
break;
}
}
}
int main()
{
int minHeap[5];
int k = 5;
FILE* fout = fopen("data.txt", "r");
if (fout == NULL)
{
perror("fopen fail");
return;
}
for (int i = 0; i < 5; ++i)
{
fscanf(fout, "%d", &minHeap[i]);
}
// 建小堆
for (int i = (k - 1 - 1) / 2; i >= 0; --i)
{
AdjustDown(minHeap, k, i);
}
int val = 0;
while (fscanf(fout, "%d", &val) != EOF)
{
if (val > minHeap[0])
{
minHeap[0] = val;
AdjustDown(minHeap, k, 0);
}
}
for (int i = 0; i < k; ++i)
{
printf("%d ", minHeap[i]);
}
printf("\n");
fclose(fout);
return 0;
}
运行结果：
