2021-09-08

欢迎来到初阶数据结构，本篇文章带你理解堆和堆排序

初阶数据结构-堆排序

堆的概念及结构

1. 什么是堆？
   如果有一个关键码的集合K = {k0，k1， k2，…，kn-1}，把它的所有元素按完全二叉树的顺序存储方式存储在一个一维数组中，并满足：Ki <= K2i+1 且 Ki<= K2i+2 (Ki >= K2i+1 且 Ki >= K2i+2) i = 0，1，2…，则称为小堆(或大堆)。将根节点最大的堆叫做最大堆或大根堆，根节点最小的堆叫做最小堆或小根堆。
2. 堆的性质
   堆中某个节点的值总是不大于或不小于其父节点的值；
   堆总是一棵完全二叉树。
   下面直接上图:
   
   这里我们可以发现不管是大小根堆它的最值都是在第一个，这样我们就可以对数据排序。如果给了我们一组数据怎么排序呢？首先就是建堆，那怎么建堆呢？接下来看一组数据

int a[] = {27,15,19,18,28,34,65,49,25,37};

堆的向下调整算法

这时就引入向下调整算法
向下调整算法有一个前提：左右子树必须是一个堆，才能调整。

转换成代码：

void AdjustDown(int* a, int n, int parent)
{
	int child = parent * 2 + 1;
	while (child < n)
	{
		// 选出左右孩子中小/大的那个
		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;
		}
	}
}

3.那我们给一组不满堆的数组呢？

int a[] = { 27, 37, 28, 18, 19, 34, 65, 25, 49, 15 };

我们可以发现这组数据不满足大/小根堆，那该怎么处理呢？

堆排序的实现

4.来测试建没建成大根堆

void HeapSort(int *a, int n)
{
	for (int i = (n-1-1) / 2; i >= 0; --i)
	{
		AdjustDown(a, n, i);
	}
}
int main()
{
	int a[] = { 27, 37, 28, 18, 19, 34, 65, 25, 49, 15 };
	

	HeapSort(a,sizeof(a) / sizeof(int));

	for (int i = 0; i < sizeof(a) / sizeof(int); i++)
	{
		printf("%d ", a[i]);
	}
	printf("\n");
	return 0;
}

大根堆就建好了
小堆如下图

接下就可以排序了。如果排升序就建大根堆，排降序建小根堆。为什么呢？

下面是完整代码：

#include<stdio.h>
void Swap(int *px,int *py)
{
	int tmp = *px;
	*px = *py;
	*py = tmp;
}
//降序  建小堆
//升序  建大堆
void AdjustDown(int* a, int n, int parent)
{
	int child = parent * 2 + 1;
	while (child < n)
	{
		// 选出左右孩子中小/大的那个
		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;
		}
	}
}

void HeapSort(int *a, int n)
{
	for (int i = (n-1-1) / 2; i >= 0; --i)
	{
		AdjustDown(a, n, i);
	}

	// O(N*logN)
	//最后一个数不看成堆里的，继续向下调整
	int end = n - 1;
	while (end > 0)
	{
		Swap(&a[0], &a[end]);
		AdjustDown(a, end, 0);
		--end;
	}
	
}

int main()
{
	int a[] = { 27, 37, 28, 18, 19, 34, 65, 25, 49, 15 };
	

	HeapSort(a,sizeof(a) / sizeof(int));

	for (int i = 0; i < sizeof(a) / sizeof(int); i++)
	{
		printf("%d ", a[i]);
	}
	printf("\n");
	return 0;
}

升序就排好了。
本篇文章到这里就已经结束了。请大家多多指教！