数据结构之堆的实现

堆是一种常用的数据结构，逻辑结构是完全二叉树，而物理存储结构是用顺序表方式存储的，下面我们来看一下它的结构和实现；

1.堆的顺序表存储

typedef int HPDataType;
typedef struct Heap
{
	HPDataType* a;
	int size;
	int capacity;
}HP;

2.堆的初始化

void HeapInit(HP* php)
{
	assert(php);

	php->a = (HPDataType*)malloc(sizeof(HPDataType) * 4);
	if (php->a == NULL)
	{
		perror("malloc fail");
		return;
	}

	php->size = 0;
	php->capacity = 4;
}

顺序表掌握了写这东西根本就不是事情；

3.堆的销毁

void HeapDestroy(HP* php)
{
	assert(php);
	free(php->a);
	php->capacity = 0;
	php->size = 0;
}

4.堆的插入

在堆中尾插一个元素是没有意义的，这里的插入是要在插入一个元素之后仍保持堆的结构；

void HeapPush(HP* php, HPDataType x)
{
	assert(php);

	if (php->size == php->capacity)
	{
		HPDataType* tmp = (HPDataType*)realloc(php->a, sizeof(HPDataType) * php->capacity * 2);
		if (tmp == NULL)
		{
			perror("realloc fail");
			return;
		}
		php->a = tmp;
		php->capacity *= 2;
	}

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

	AdjustUp(php->a, php->size - 1);
}

那么怎么样才能使它仍保持堆的结构呢，这就需要对我们最后插入的那个元素进行向上调整；

5.向上调整算法

void AdjustUp(HPDataType* a, int child)
{
	int parent = (child - 1) / 2;
	//while (parent >= 0)
	while (child > 0)
	{
		if (a[child] > a[parent])
		{
			Swap(&a[child], &a[parent]);
			child = parent;
			parent = (child - 1) / 2;
		}
		else
		{
			break;
		}
	}
}

child=parent,parent=(child-1/2)进行迭代，当parent<0的时候终止；

6.堆的删除；

堆的删除是删除顶部的元素才是有意义的，如果直接对堆顶部的元素进行覆盖，不仅效率低下，而且会做成父子关系紊乱，最好的方法是将顶部元素与尾部元素互换，，删除尾部元素然后对顶部元素向下调整；

void HeapPop(HP* php)
{
	assert(php);
	assert(!HeapEmpty(php));

	// 删除数据
	Swap(&php->a[0], &php->a[php->size - 1]);
	php->size--;

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

7.堆的向下调整；

void AdjustDown(HPDataType* a, int n, int parent)
{
	int child = parent * 2 + 1;
	while (child < n)
	{
		// 选出左右孩子中大的那一个
		if (child + 1 < n && a[child + 1] > a[child])
		{
			++child;
		}

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

8.堆的判空

bool HeapEmpty(HP* php)
{

	assert(php);
	return php->size == 0;
}

9.堆的数据元素个数；

int HeapSize(HP* php)
{
	assert(php);
	return php->size;
}

好了堆的实现就到这里了，下次再来探讨堆排序以及Topk问题把！