【数据结构】建堆、堆的向下调整及其复杂度、堆的向上调整及其复杂度、Top-K问题

一.完全二叉树的顺序结构

堆的逻辑结构采用完全二叉树，而堆就是在一定条件下将完全二叉树使用数组存储，我们可以利用完全二叉树更好的学习堆。

普通的二叉树是不适合用数组来存储的，因为可能会存在大量的空间浪费。而完全二叉树更适合使用顺序结构（数组）存储。

如上图该完全二叉树（按从上到下，从左到右顺序存储）的数组结构就叫做堆（上图的数组满足条件所以可以叫做堆，条件下面会讲）

• 需要注意，这里的堆和操作系统虚拟进程地址空间中的堆是两回事，一个是数据结构，一个是操作系统中管理内存的一块区域分段。

二.堆的概念及结构

那是不是只要是一个完全二叉树，把它按顺序放在数组里就是一个堆呢？当然不是

堆的种类分为两种：

1. 一个堆中某个节点的值总是不大于其父节点的值叫：大堆
2. 一个堆中某个节点的值总是不小于其父节点的值叫：小堆

我们得出堆的性质：

• 堆中某个节点的值总是不大于或不小于其父节点的值；
• 堆总是一颗完全二叉树。（可以在逻辑上形成完全二叉树）

只有满足这两点的数组才可以被称之为堆

三.堆的实现

现在给出一个数组，逻辑上看做是一颗完全二叉树。我们已经知道堆分为大堆和小堆两种，那我们要如何才能将它变成一个完整的堆呢？

int arr[8] = { 7,4,2,9,3,4 };

• 堆的数据结构，难点就在于堆的创建、插入和删除，所以这三个功能会放在一起讲，剩余的功能最后讲解。

这里就需要先来学一种方法完成堆的构建：堆向下调整

1.堆向下调整

按照完全二叉树的逻辑结构，从最后一个节点的父节点开始，以该节点下标为起始，与自己的两个孩子节点（可能为一个孩子节点）比较大小并根据情况交换位置（大堆比较更大的孩子节点，小堆比较更小的孩子节点），一直调整到根节点，最终得到大堆/小堆

• 向下调整算法的前提：根节点的左右子树必须都为堆，才能调整，否则在本就无序的情况下去，去使用根节点的值去比对查找，是无法找到适合的元素的，所以向下调整只能从最后一个节点的父节点开始（也可以说是倒数的第一个非叶子节点）对比，从后向前依次建堆，才能保证每个结点在进行向下调整时左右子树都为堆。

我们在这些做一个大堆并画出它的流程图：

• 从最后一个父节点向下比较，建大堆，将最大值移到上面，数组的切换如上图

• 接着使用下一个父节点与其子节点中最大值比较，与最大值交换

• 在将下一个父节点与其子节点中的最大值比较并交换，查看交换后的子节点是否为其子节点的最大值，若不是则继续进行向下调整，否则以调整到根节点，建堆结束。

我们之前以及提过了，这个堆的存储形式是一个数组，而最后一个节点就是数组下标最大的元素，那我们要如何求出它的父节点？

这个是有固定的公式，根据这个公式：
当我们知道孩子节点的下标就能求出其父节点的下标，知道父亲节点的下标，就能求出孩子节点的下标，

下标如下图：

• 知道父亲节点下标，求孩子节点下标
  father = 0 ：
  左孩子 = father * 2 + 1 = 1；
  右孩子 = father * 2 + 2 = 2；
  father = 3 ：
  左孩子 = father * 2 + 1 = 7；
  右孩子 = father * 2 + 2 = 8；
  知道父亲节点，求孩子节点的公式如下：
  左孩子 = father * 2 + 1；
  右孩子 = father * 2 + 2；

• 知道孩子节点下标，求父亲节点下标
  知道左孩子下标
  由上面的公式推导：father = (左孩子-1)/2;
  知道右孩子下标
  由上面的公式推导：father = (右孩子-2)/2;
  其中，左孩子一定是奇数，右孩子是偶数，而在C语言中，一个偶数减1后除以2和一个偶数减2后除以2是相同的，因为下标为整数，偶数减1后除以2所得必须为整数，除不尽的部分直接舍弃。
  因此，我们不需要知道该孩子节点是左孩子还是右孩子。
  知道孩子节点下标，求父亲节点下标公式如下：
  father = (孩子-1)/2；

2.向下调整建堆

我们已经知道了一个堆是如何创建的，现在有如下数组，我们使用代码来实现堆的创建。

int arr[8] = { 7,4,2,9,3,4,5,6 };

代码如下：

void swap(int* a1, int* a2)
{
	int tmp = *a1;
	*a1 = *a2;
	*a2 = tmp;
}

void AdjustDown(int* arr, int n, int father)//向下调整
{
	int child = father * 2 + 1;//左孩子节点下标

	while (child < n) //向下调整，child只会越来越大，当超出数组范围时，退出循环
	{
		if (child+1 < n && arr[child] < arr[child + 1])//判断是否存在右孩子节点，存在左右孩子结点谁大，要是建小堆，取小的那个元素
			child = child + 1;
		if (arr[father] < arr[child])//判断父亲结点和孩子结点那个大，若孩子结点大，则进行交换，并继续向下调整，建小堆取小值即可
		{
			swap(&arr[father], &arr[child]);//两节点进行交换
			father = child;//孩子结点的下标赋给父亲结点
			child = father * 2 + 1;//孩子节点获得新的左孩子下标
		}
		else
			break;
	}
}

int CreatHeap(int* arr, int n)//arr为要建堆的数组
{
	for (int i = (n - 1 - 1) / 2; i >= 0; i--)//循环的初始值为倒数第一个非叶子节点的下标
	{
		AdjustDown(arr, n, i);//向下调整
	}
}

3.向下调整建堆时间复杂度

时间复杂度是求该函数在最坏的情况下基本操作次数，而我们现在算的是建堆这一功能的时间复杂度，因为堆是完全二叉树，而满二叉树（除叶子节点外，每个节点都有两个孩子节点）也是完全二叉树，此处我们就能使用满二叉树来判断堆的时间复杂度。

因为需要按照最坏的情况考虑，我们就将堆中除最后一层外的其他节点都需要向下移动到最底层。

所以：建堆的时间复杂度为O(N)

4.堆的插入（向上调整）

如下图所示，我们在原有完全二叉树的基础上，向其中尾插一个10，之后就需要进行向上调整，做法如下图：

该堆为小堆，先找到新插入节点的父节点，进行对比，父节点大，进行替换，新插入的节点取代父节点，并向新的父节点进行对比操作，直到到达根节点。若是父节点小于新插入的节点，停止操作，表面堆的插入以完成。

注意：这里最好不要用向下调整，堆的插入操作是在一个堆已经成型的情况下进行，如果用向下调整，新插入的节点的父节点后的所有节点都要重新遍历一遍，时间复杂度一定要比只是移动新节点的向上调整大。

知道孩子节点，如何求父节点的公式之前已经讲了，这里我们就能直接得出代码（这里做的是小堆的插入）：

向上调整代码如下：

void AdjustUp(int* arr, int child)
{
	int father = (child - 1) / 2;//获得父亲节点的下标

	while (child)
	{
		if (arr[father] > arr[child])//孩子结点数据小于父亲结点数据，两数交换
		{
			swap(&arr[father], &arr[child]);
			child = father;
			father = (child - 1) / 2;
		}
		else
			break;
	}
}

• 堆中的结点个数为：n = h^2 - 1;
  堆中的层数为：h = log(n+1);
  堆插入的时间复杂度为堆的层数，时间复杂度为：O(logN)
• 如果使用向下调整完成堆插入，相当于在经历一遍建堆，因为堆已经建好，所以比较的次数相比初次建堆要小，但终归没有直接向上排序来的快。

5.向上调整建堆

我们在堆的插入中已经学了堆的向上调整的思想，我们也可以利用向上调整来建堆，但是这个时间复杂度要大于向下调整，实际操作中不会用到这个方法建堆，这里简单介绍一下。

• 使用如上图片中的数组建大堆

• 向上调整建堆，需要按照数组一个一个插入堆中比较，出现比父节点大的数据，进行向上调整，直到数组全部进入堆中。

• 发现插入数据比父节点大，按照向上调整，直到遇到更大的或到达根节点停止。
  
  代码如下：

void AdjustUp(int* arr, int child)
{
	int father = (child - 1) / 2;//获得父亲节点的下标

	while (child)
	{
		if (arr[father] > arr[child])//孩子结点数据小于父亲结点数据，两数交换
		{
			swap(&arr[father], &arr[child]);
			child = father;
			father = (child - 1) / 2;
		}
		else
			break;
	}
}
void CreatHeap(int* a,int n)
{
	for (int i = 0; i < n; i++)
	{
		AdjustUp(a, i);
	}
}

6.向上调整建堆时间复杂度

以最坏情况来算时间复杂度，与向下调整不同，假设高度为h，最底层每个节点都要向上移动h-1次，剩余节点按层数依次类推。

将这些移动次数相加为：

所以：建堆的时间复杂度为O(Nlog(N))

向上调整建堆的时间复杂度要大于向下调整建堆

7.堆的删除

堆的删除就是将堆顶的数据删除。

具体操作：将堆顶的数据与最后一个节点的数据交换，然后删除数组中的最后一个数据，在对堆顶的数据进行向下调整，如下图。

8.向下调整和向上调整的使用类型

我们已经知道了向上调整和向下调整的方式及其时间复杂度，现在我们来看一下，在什么情况下，应用这两种方式最为合理。

向下调整：

• 当给出一个无序的数组，要求我们建堆时，我们将该数组看作一个整体，此时采用向下调整建堆
• 当我们进行堆的删除时，因为将最后一个元素放在了堆首，此时采用向下调整

向上调整：

• 当我们将无序数组一个元素一个元素插入堆中建堆时，我们使用向上调整建堆的方式
• 在进行堆插入操作时，采用向上调整

9.堆的代码实现

由上面几个功能，比如堆的插入和删除，这是要直接对数组进行扩容等相关操作的，如果只在数组上进行会造成很多的麻烦，所以我们需要创建一个结构体来用来存储堆，而堆的删除也用用到了结构体相关的内容，结构体定义如下：

typedef int HPDataType;
typedef struct Heap
{
HPDataType* _a;
int _size;
int _capacity;
}Heap;

堆的其他功能实现难度不大，这里就只整体展示代码，不一个个讲解了。
注意：下面代码中一些类型用自定义结构体类型交换，与上面实现的代码相同
堆的所有功能代码如下;

//打印堆
void PrintHeap(Heap* hp)
{
	assert(hp);
	for (int i = 0; i < hp->size; i++)
	{
		printf("%d ", hp->a[i]);
	}
	printf("\n");
}

void HeapInit(Heap* hp)
{
	hp->a = NULL;
	hp->capacity = 0;
	hp->size = 0;
}

// 堆的销毁
void HeapDestory(Heap* hp)
{
	assert(hp);
	free(hp->a);
	hp->a = NULL;
	hp->capacity = 0;
	hp->size = 0;
}

堆的构建——偷懒的写法——使用向上调整实现堆的构建
//void HeapCreat(Heap* hp, HPDataType* a, int n)
//{
//	hp->a = (HPDataType*)malloc(sizeof(HPDataType) * n);
//	if (!hp->a)
//	{
//		perror("malloc fail!");
//		exit(-1);
//	}
//	hp->capacity = n;
//
//	for (int i = 0; i < n; i++)
//	{
//		HeapPush(hp, a[i]);
//	}
//}

//堆的构建——建堆算法
void HeapCreat(Heap* hp, HPDataType* a, int n)
{
	hp->a = (HPDataType*)malloc(sizeof(HPDataType) * n);
	if (!hp->a)
	{
		perror("malloc fail!");
		exit(-1);
	}

	memcpy(hp->a, a, sizeof(HPDataType) * n);
	hp->capacity = hp->size = n;

	//建堆算法——函数接收父亲结点算法
	for (int i = (n - 1 - 1) / 2; i >= 0; i--)
	{
		AdjustDown(hp->a, n, i);
	}
}

//两数交换
void swap(HPDataType* a1, HPDataType* a2)
{
	HPDataType tmp = *a1;
	*a1 = *a2;
	*a2 = tmp;
}

//向上调整
void AdjustUp(HPDataType* arr, int child)
{
	int father = (child - 1) / 2;//父亲结点位置

	while (child)
	{
		if (arr[child] > arr[father])//孩子结点数据大于父亲结点数据，两数交换
		{
			swap(&arr[child], &arr[father]);
			child = father;
			father = (child - 1) / 2;
		}
		else
			break;
	}
}

// 堆的插入
void HeapPush(Heap* hp, HPDataType x)
{
	assert(hp);
	if (hp->capacity == hp->size)
	{
		int newSize = hp->capacity * 2;
		HPDataType* newArr = (HPDataType*)realloc(hp->a, newSize);
		if (!newArr)
		{
			perror("realloc fail!");
			exit(-1);
		}
		hp->capacity = newSize;
		hp->a = newArr;
	}

	hp->a[hp->size++] = x;

	AdjustUp(hp->a, 0);
}

//向下调整
void AdjustDown(HPDataType* arr,int n,int parent)
{
	int child = parent * 2 + 1;

	while (child < n)
	{
		if (child + 1 < n && arr[child] > arr[child + 1])
			child = child + 1;

		if (arr[parent] > arr[child])
		{
			swap(&arr[parent], &arr[child]);
			parent = child;
			child = parent * 2 + 1;
		}
		else
			break;
	}
}

// 堆的删除
void HeapPop(Heap* hp)
{
	assert(hp);
	assert(hp->size);

	hp->a[0] = hp->a[--hp->size];

	AdjustDown(hp->a, hp->size, 0);
}

//取堆顶的数据
HPDataType HeapTop(Heap* hp)
{
	assert(hp);
	assert(hp->size > 0);

	return hp->a[0];
}

//堆的数据的个数
int HeapSize(Heap* hp)
{
	assert(hp);

	return hp->size;
}

//判空
int HeapEmpty(Heap* hp)
{
	assert(hp);

	return hp->size == 0;
}

四.Top-K问题

问题：在一个很大的数据中，取出它最大或最小的前k个值。

比如：专业前10名、世界500强、富豪榜、游戏中前100的活跃玩家等。

对于这个问题，想到最简单的方法就是排序，但是数据量非常大，当数据不可能一下子全部加载到内存中时，排序就不太可取了。

最佳的方法就是用堆来解决，基本思路如下：

1. 用数据集合中前k个元素来建堆

• 取前k个最大的元素，建小堆
• 取前k个最小的元素，建大堆

2.用剩余的N-k个元素依次与堆顶元素来比较，不满足则替换堆顶元素，在进行向下调整。

将剩余N-k个元素依次与堆顶元素比较完之后，堆中剩余的K个元素就是所求的前K个最小或者最大的元素。

如下图，为取前6个最大的元素的一次对比图：

具体代码如下：

void TopK(int* arr,int n,int k)
{
	int* tmpArr = (int*)malloc(sizeof(int) * k);
	
	//方法1
	//memcop(tmpArr, arr, sizeof(int) * k);//使用内存函数拷贝数组前k个数

	//方法2
	for (int i = 0; i < k; i++)
	{
		tmpArr[i] = arr[i];
	}

	for (int i = (k - 1 - 1) / 2; i >= 0; i--)
	{
		AdjustDown(tmpArr, k, i);//向下调整建堆
	}
	for (int i = k; i < n; i++)
	{
		if (arr[i] > tmpArr[0])
		{
			swap(&arr[i], &tmpArr[0]);
		    AdjustDown(tmpArr, k, 0);//将此时栈顶的元素进行向下调整
		}
	}
	for (int i = 0; i < k; i++)
	{
		printf("%d ", tmpArr[i]);//打印出这k个数，但此时这k个数是无序的
	}
}

五.堆排序

当我们掌握了上面的堆的创建、向下调整和父亲与孩子节点下标的转换后，堆排序的实现就非常简单

先说一下思路：

当我们想要得到一个排好序的数组时，我们需要将它先建成堆。

• 升序：大堆
• 降序：小堆
  假设我们要对一个大小为n的数组进行降序，
  首先，建好的小堆的堆顶就是最小的数，我们将它和第n个数交换，在将堆的大小调整为（n-1），对堆顶进行向下调整。
  其次，此时堆顶的数就是次小的数，我们在将它和第n-1个数交换，在将堆的大小调整为（n-2），对堆顶进行向下调整。
  如此循环，最后得到的n个数就是降序排好的

如下图，是堆排序的部分循环图：

剩余步骤与上述步骤相同，不在展示。

根据该思路，堆排序的实现代码如下：

void AdjustDown(int* arr, int n, int father)
{
	int child = father * 2 + 1;//左孩子节点下标

	while (child < n)
	{
		if (child+1 < n && arr[child] > arr[child + 1])//判断是否存在右孩子节点，存在左右孩子结点谁大
			child = child + 1;
		if (arr[father] > arr[child])//判断父亲结点和孩子结点那个大，若孩子结点大，则进行交换，并继续向下调整
		{
			swap(&arr[father], &arr[child]);//两节点进行交换
			father = child;//孩子结点的下标赋给父亲结点
			child = father * 2 + 1;//孩子节点获得新的左孩子下标
		}
		else
			break;
	}
}

int* HeapSort(int* arr,int n)
{
	for (int i = (n-1-1)/2; i >= 0; i--)//先建堆
	{
		AdjustDown(arr, n, i);
	}

	for (int i = 0; i < n; i++)//向下调整堆排序
	{
		swap(&arr[i], &arr[n - i - 1]);//堆顶元素与最后一个元素交换，将最大值或最小值放在最后
		AdjustDown(arr, n - i - 1, 0);
	}

	return arr;
}

时间复杂度： O(N*log(N))
空间复杂度： O(1)
稳定性： 不稳定

六.总结

1. 向下调整建堆的时间复杂度小于向上调整建堆，建堆使用向上调整。
2. 向下调整要求目标节点之下是一个完整堆，向上调整要求目标节点上是一个完整堆。
3. 堆删除是将首元素和尾元素置换，总元素各数减1，对新的首元素进行向下调整。
4. 堆插入时将新的元素插入尾部，堆新的元素进行向上调整。
5. Top-k问题是限定堆整体的大小，对堆的头节点进行操作
6. 堆排序首元素依此于尾元素替换，并缩小堆的大小，当堆中元素个数为0时，堆排序完成，堆排序与堆删除十分相似。