数据结构--堆

一、堆的概念

堆(Heap)是计算机科学中一类特殊的数据结构的统称。

堆通常是一个可以被看做一棵完全二叉树的数组对象。

堆满足下列性质：

• 堆中某个节点的值总是不大于或不小于其父节点的值。
• 堆总是一棵完全二叉树。

大根堆：每个根节点都不小于子节点的堆叫大根堆。

小根堆：每个根节点都不大于子节点的堆叫小根堆。

二、堆的存储结构

堆是一个基于二叉树的数据结构，可以用数组进行存储。

                                                                 逻辑结构

                                                                存储结构

1.对于任意节点i，它的左孩子节点为2*i+1，右孩子节点为2*i+2，父节点为(i-1)/2。

2.要注意，大根堆不是升序，小根堆不是降序。

三、堆的实现

3.1向上调整法

一般用于新节点的插入。

步骤(以小根堆为例)：

1、将新节点插入到堆的末尾。

2、将新节点的值与父节点的值进行比较。

3、如果新节点的值小于父节点的值，将新节点的值与父节点的值交换。

4、继续向上进行比较，重新步骤3。

5、如果新节点的值不小于父节点的值，则不再比较，跳出循环。

代码：

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;
		}
	}
}

3.2向下调整法

一般用于节点的删除。

步骤(以小根堆为例)：

1、比较根节点的两个子节点，选出较小的子节点。

2、比较根节点与较小子节点。

3、如果根节点大于子节点，交换根节点与子节点。

4、重复进行以上步骤，直到根节点不再大于子节点。

代码：

void AdjustDown(HPDataType* a, int n, int parent)//
{
	int child = parent * 2 + 1;
	//假设左孩子为小的
	while(child < n)
	{
		if (child + 1 < n && a[child] > a[child + 1])
		{
			child++;
		}

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

}

3.3插入数据

先检查数组大小是否足够，如果空间不够插入，则增长数组。然后用向上调整法调整堆。

代码：

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

	if (php->capacity == php->size)
	{
		int newcapacity = php->capacity == 0 ? 4 : 2 * php->capacity;
		HPDataType* tmp = (HPDataType*)realloc(php->a, newcapacity * sizeof(HPDataType));
		if (tmp == NULL)
		{
			perror("fail");
			return;
		}
		php->a = tmp;
		php->capacity = newcapacity;
	}
	php->a[php->size] = x;
	php->size++;

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

}

3.4删除

删除数据并不是删除最后一个数据，而是删除堆顶数据。

步骤：

1、交换堆顶节点和最后一个节点。此时，最后一个节点就变成了原先的堆顶节点，堆顶节点就变成了原先的最后一个节点。

2、删除最后一个节点(也就是原先的堆顶节点)。

3、对堆顶节点(也就是原先的最后一个节点)使用向下调整法。

代码：

void HPPop(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);
}

四、完整代码

typedef int HPDataType;

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

void Swap(HPDataType* p1, HPDataType* p2);

//初始化
void HPInit(HP* php);
//销毁
void HPDestroy(HP* php);
//插入
void HPPush(HP* php, HPDataType x);

//向上调整法
void AdjustUp(HPDataType* a, int n);

//删除
void HPPop(HP* php);

//向下调整法
void AdjustDown(HPDataType* a, int n, int parent);

//取堆顶数据
HPDataType HPTop(HP* php);
//判断堆是否为空
bool HPEmpty(HP* php);

//初始化
void HPInit(HP* php)
{
	assert(php);

	php->a = NULL;
	php->capacity = php->size = 0;
}

//销毁
void HPDestroy(HP* php)
{
	assert(php);

	free(php->a);
	php->a = NULL;
	php->capacity = php->size = 0;
}

void Swap(HPDataType* p1, HPDataType* p2)
{
	HPDataType tmp = *p1;
	*p1 = *p2;
	*p2 = tmp;
}

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;
		}
	}
}

//插入
void HPPush(HP* php, HPDataType x)
{
	assert(php);

	if (php->capacity == php->size)
	{
		int newcapacity = php->capacity == 0 ? 4 : 2 * php->capacity;
		HPDataType* tmp = (HPDataType*)realloc(php->a, newcapacity * sizeof(HPDataType));
		if (tmp == NULL)
		{
			perror("fail");
			return;
		}
		php->a = tmp;
		php->capacity = newcapacity;
	}
	php->a[php->size] = x;
	php->size++;

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

}

void AdjustDown(HPDataType* a, int n, int parent)//n 是size的大小
{
	int child = parent * 2 + 1;
	//假设左孩子为小的
	while(child < n)
	{
		if (child + 1 < n && a[child] > a[child + 1])
		{
			child++;
		}

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

}

//删除
void HPPop(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);
}

HPDataType HPTop(HP* php)
{
	assert(php);
	assert(php->size > 0);
	return php->a[0];
}
bool HPEmpty(HP* php)
{
	assert(php);
	return php->size == 0;
}