堆-----数据结构

最新推荐文章于 2024-11-09 03:58:56 发布

不是颜

最新推荐文章于 2024-11-09 03:58:56 发布

阅读量419

点赞数

文章标签：数据结构 c++ 算法

本文链接：https://blog.csdn.net/qq_66805048/article/details/133912189

版权

数据结构专栏收录该内容

16 篇文章 0 订阅

订阅专栏

引言

什么是堆？堆是一种特殊的数据结构（用数组表示的树）。

为什么要使用到堆？比如一场比赛，如果使用擂台赛的方式来决出冠军（实力第一），就很难知道实力第二的队伍是什么了。

但是下图能很清楚的表示各队伍的强弱关系。

堆的特点

上图就是一个最大堆，解释：每一个圆都是一个节点，数字代表着键值，其中95是93和92的父节点，93和92是95的子节点，93和92是兄弟节点（父节点为同一个），根节点就是键值最大的节点，为95，最后一个节点是87，最后一个左子节点也是87。

最大堆的特点：

满足以下三点

1.每个节点最多可以有两个子节点。

2.根节点的键值是所有堆节点键值中最大者，且每个节点的值都大于其子（孩子）节点。

3.除了根节点没有兄弟节点，最后一个左子节点可以没有兄弟节点，其他节点必须有兄弟节点。

最好是自己理解，不用强记。其中有一点要注意：A的兄弟节点的子节点可能大于A，相当于在比赛中，其中一个小组都是弱队，那么一个弱队却可以闯入半决赛一样。

最小堆的特点的话就将第二点中的大改为小就可以了，其他的特点一样。这里讨论的是最大堆

数组形式表示

父节点和子节点的关系：

i 的左子节点：2i+1 ，右子节点：2i+2

i 的父节点：（i-1）/2

i 是从 0 开始的

再将上图的堆转换为数组的形式，如下图：

这就是一个最大堆的数组表示形式。

在数组中快速创建堆

也就是怎么把任意一个数组变成最大/小堆。

我们先把堆的最小单位拿出来，如下图：

他不是一个最大堆，如果要变成最大堆，只需要父节点是95，子节点分别是86、88就可以了（86和95交换）。而一个最大堆是由若干个这个最小单位组成的，所以第一步就是将所有的堆的最小单位变成最大堆。

1、首先先找到最后一个节点的父节点，找出该节点的最大子节点与自己比较，如果大于自身，就交换两个节点。

2、继续移动到上一个父节点（也就是下标 -1 的地方），重复做第一步的比较操作，直到遍历所有的父节点。

当我们移动完所有的父节点，那最大堆就形成了吗？还疏忽了一个地方。例如当移动到某个父节点时，如下图：

最开始父节点是88，与子节点95交换了，那父节点就是95，95 的子节点就是 88，那88一定大于他的子节点吗？很显然这个答案是不一定，因为 88 的子节点只满足小于之前的父节点 95，所以还需要向下调整，直到子节点都小于父节点。

3、对每次移动中，变成子节点的节点，向下调整，也就是判断他与子节点是否满足最大堆的特点，不满足还要继续移动节点（向下调整），满足的话就接着下个父节点。

4、所有的节点交换完毕，最大堆构建完成。

堆的算法实现

堆数据结构的定义

#define DEFAULT_CAPCITY 120 //默认的堆容量

typedef struct _Heap
{
	int* arr;		//存储堆元素的数组
	int size;		//堆中元素的个数
	int capcity;	//堆的容量
}Heap;

//函数声明
void buildHeap(Heap& heap);
bool inset(Heap& heap, int value);
bool initHeap(Heap& heap, int* orginal, int size);
void adjustDown(Heap& heap, int i);
void adjustUp(Heap& heap, int i);

堆的初始化

bool initHeap(Heap& heap,int *orginal,int size) 
{
    //orginal 是指向数组的指针，而这个数组是我们要传入堆的数组

	int capcity = DEFAULT_CAPCITY > size ? DEFAULT_CAPCITY : size; //取size和默认容量的最大值
	
	heap.arr = new int[capcity];
	if (!heap.arr) return false; //申请失败

	heap.capcity = capcity;

	if (size > 0) //size合法
	{
		memcpy(heap.arr, orginal, sizeof(int) * size);
		heap.size = size;
		//建堆
		buildHeap(heap);
	}

	return true;
}

堆的创建

//建堆，从最后一个父节点逐个向前调整所有的父节点（直到根节点），确保每一个父节点都是一个最大堆
//那么，整体上就是一个最大堆
void buildHeap(Heap& heap)
{
	int i = (heap.size - 2) / 2; //因为下标从0开始，heap.size-1就得到下标，再结合公式就是这个式子

	for (; i >= 0; i--)
	{
		adjustDown(heap, i); //向下调整包含了构建最大堆，如果感到困惑，先看向下调整函数
	}
}

堆的向下调整函数

void adjustDown(Heap& heap, int i)
{
	int temp = heap.arr[i]; //保存父节点的键值
	int parent = 0 ,child = 0;

	for (parent = i; (2 * parent + 1) < heap.size; parent = child) 
	{
		child = 2 * parent + 1; //先指向左子节点
		
		//指向两个子节点中最大的节点
		if (child + 1 < heap.size && heap.arr[child] < heap.arr[child + 1])
		{
			child = child + 1;
		}
        
		if (temp >= heap.arr[child])
		{
			break; //无需向下调整
		}
		else
		{
			heap.arr[parent] = heap.arr[child];
			heap.arr[child] = temp;
		}
		
	}
}

堆的插入新元素

1、插入新的元素到最大堆的尾部，也就是数组的后面

2、插入的元素可能会破环这个最大堆，需要重新调整，和父节点比较，如果比父节点大，就交换两个节点……重复直到新节点比父节点小或者新节点变为根节点（调整到位）。

设计两个函数，一个是插入，一个是向上调整。

bool insert(Heap& heap, int value)
{
	if (heap.size == heap.capcity) //堆空间不足
	{
		return false;
	}

	int i = heap.size ; //指向新加元素的下标
	heap.arr[heap.size++] = value;
	adjustUp(heap , i);
	return true;
}

void adjustUp(Heap& heap, int i)
{
	if (i <= 0 || i >= heap.size) return;

	while (i > 0)
	{
		int parent = (i - 1) / 2;
		if (parent >= 0) // 父节点没越界
		{
			if (heap.arr[parent] < heap.arr[i])
			{
				int temp = heap.arr[i];
				heap.arr[i] = heap.arr[parent];
				heap.arr[parent] = temp;
				i = parent;
			}
			else
			{
				break; //无需调整
			}
		}
		else
		{
			break; //父节点出界
		}
	}
}

看到这，你会发现堆的创建还有一种方式，也就是将数组的元素一个一个插入，也能得到最大堆。

源代码

#include <iostream>

using namespace std;




#define DEFAULT_CAPCITY 120 //默认的堆容量

typedef struct _Heap
{
	int* arr;		//存储堆元素的数组
	int size;		//堆中元素的个数
	int capcity;	//堆的容量
}Heap;

void buildHeap(Heap& heap);
bool insert(Heap& heap, int value);
bool initHeap(Heap& heap, int* orginal, int size);
void adjustDown(Heap& heap, int i);
void adjustUp(Heap& heap, int i);

//初始化堆
bool initHeap(Heap& heap,int *orginal,int size) 
{
    //orginal 是指向数组的指针，而这个数组是我们要传入堆的数据

	int capcity = DEFAULT_CAPCITY > size ? DEFAULT_CAPCITY : size; //取size和默认容量的最大值
	
	heap.arr = new int[capcity];
	if (!heap.arr) return false;

	heap.capcity = capcity;

	if (size > 0)
	{
		memcpy(heap.arr, orginal, sizeof(int) * size);
		heap.size = size;
		//建堆
		buildHeap(heap);
	}

	return true;
}

//建堆，从最后一个父节点逐个向前调整所有的父节点（直到根节点），确保每一个父节点都是一个最大堆
//那么，整体上就是一个最大堆
void buildHeap(Heap& heap)
{
	int i = (heap.size - 2) / 2; //因为下标从0开始，heap.size-1就得到下标

	for (; i >= 0; i--)
	{
		adjustDown(heap, i);
	}
}

void adjustDown(Heap& heap, int i)
{
	int temp = heap.arr[i]; //父节点的键值
	int parent = 0 ,child = 0;

	for (parent = i; (2 * parent + 1) < heap.size; parent = child)
	{
		child = 2 * parent + 1;
		
		//指向两个子节点中最大的节点
		if (child + 1 < heap.size && heap.arr[child] < heap.arr[child + 1])
		{
			child = child + 1;
		}

		if (temp >= heap.arr[child])
		{
			break; //无需向下调整
		}
		else
		{
			heap.arr[parent] = heap.arr[child];
			heap.arr[child] = temp;
		}
		
	}
}

//堆——插入新元素
bool insert(Heap& heap, int value)
{
	if (heap.size == heap.capcity)
	{
		return false;
	}

	int i = heap.size ;
	heap.arr[heap.size++] = value;
	adjustUp(heap , i);
	return true;
}

void adjustUp(Heap& heap, int i)
{
	if (i <= 0 || i >= heap.size) return;

	while (i > 0)
	{
		int parent = (i - 1) / 2;
		if (parent >= 0) // 父节点没越界
		{
			if (heap.arr[parent] < heap.arr[i])
			{
				int temp = heap.arr[i];
				heap.arr[i] = heap.arr[parent];
				heap.arr[parent] = temp;
				i = parent;
			}
			else
			{
				break; //无需调整
			}
		}
		else
		{
			break; //父节点出界
		}
	}
}
int main(void)
{
	Heap heap;
	int orginalArr[] = { 1,2,3,87,93,82,92,86,95 };

	if (!initHeap(heap, orginalArr, sizeof(orginalArr) / sizeof(int)))
	{
		cout << "初始化堆失败！" << endl;
		exit(-1);
	}

	for (int i = 0; i < heap.size; i++)
	{
		printf("%d\n",heap.arr[i]);
	}
	puts("");
	insert(heap, 100);
	for (int i = 0; i < heap.size; i++)
	{
		printf("%d\n", heap.arr[i]);
	}
	return 0;
}