15.堆

1.堆的原理

堆(Heap)是一种数据结构,通常用于实现优先队列。堆是一种树形结构,通常由一个完全二叉树构成,因此它只有两个指针,即左子节点和右子节点。堆有两种类型:最大堆(Max Heap)和最小堆(Min Heap)。在最大堆中,父节点的值大于或等于其子节点的值;在最小堆中,父节点的值小于或等于其子节点的值。

堆的主要操作包括:

  1. 插入(Insertion):将一个元素插入到堆中。在最大堆中,插入元素时将其放入最后一个位置,然后将其与父节点比较并交换位置,直到满足堆的性质。在最小堆中,插入元素时将其放入第一个位置,然后将其与子节点比较并交换位置,直到满足堆的性质。
  2. 删除(Deletion):从堆中删除一个元素。在最大堆中,删除最后一个元素,然后将最后一个元素与第一个元素交换位置,然后将其与子节点比较并交换位置,直到满足堆的性质。在最小堆中,删除第一个元素,然后将第一个元素与最后一个元素交换位置,然后将其与父节点比较并交换位置,直到满足堆的性质。
  3. 查找(Finding):查找堆中最大或最小的元素。在最大堆中,查找最后一个元素;在最小堆中,查找第一个元素。

堆是一种高效的数据结构,可以用于实现优先队列、快速排序等算法。

最大堆特点:

1.每个节点最多可以有两个节点
2.根节点的键值是所有堆节点键值中最大者,且每个节点的值都比孩子大
3.除了根节点没有兄弟节点,最后一个左子节点可以没有兄弟节点,其他节点必须有兄弟节点

看图识最大堆: A B 不是堆,C 是最大堆

堆是你见过的最有个性的树!它是用数组表示的树

i的子节点左:2i+1  i的右子节点:2i+2
i的父节点:(i-1)/2

在数组中快速创建堆

  1. 首先我们需要找到最后一个结点的父结点如图(a),我们找到的结点是87,然后找出该结点的最大子节点与自己比较,若该子节点比自身大,则将两个结点交换。
    图(a)中,87 比左子节点95 小,则交换之。如图(b)所示

2.我们移动到前一个父结点93,如图(c)所示。同理做第一步的比较操作,结果不需要交换。

3.继续移动结点到前一个父结点82,如图(d)所示,82 小于右子节点95,则82 与95 交换,如图(e)所示,82 交换后,其值小于左子节点,不符合最大堆的特点,故需要继续向下调整,如图(f)所示

4.所有节点交换完毕,最大堆构建完成

2.堆的算法实现

堆数据结构的定义

#define DEFAULT_CAPCITY 128
typedef struct _Heap
{
    int *arr; //存储堆元素的数组
    int size; //当前已存储的元素个数
    int capacity; //当前存储的容量
}Heap;

构建最大堆

bool initHeap(Heap& heap, int* original, int size);//将数组用堆的结构存储
static void buildHeap(Heap& heap);//建成最大堆 static阻止外部访问
static void adjustDown(Heap& heap, int index);//向下调整

//初始化堆
bool initHeap(Heap& heap, int* original, int size)
{
	int capacity = DEFAULT_CAPCITY > size ? DEFAULT_CAPCITY : size;

	heap.arr = new int[capacity];
	if (!heap.arr) return false;

	heap.capacity = capacity;
	heap.size = 0;

	//如果存在原始数据则构建堆
	if (size > 0)
	{
		memcpy(heap.arr, original, size * sizeof(int));
		heap.size = size;
		buildHeap(heap);
	}
	return true;
}

/* 从最后一个父节点(size/2-1 的位置)逐个往前调整所有父节点(直到根节
点),确保每一个父节点都是一个最大堆,最后整体上形成一个最大堆*/
void buildHeap(Heap& heap)
{
	int i;
	for (i = heap.size / 2 - 1; i >= 0; i--)
	{
		adjustDown(heap, i);
	}
}

//将当前的节点和子节点调整成最大堆
void adjustDown(Heap& heap, int index)
{
	int cur = heap.arr[index];//当前待调整的节点
	int parent, child;

	/*判断否存在大于当前节点子节点,如果不存在,则堆本身是平衡的,不需要
	调整;如果存在,则将最大的子节点与之交换,交换后,如果这个子节点还有
	子节点,则要继续按照同样的步骤对这个子节点进行调整*/
	for (parent = index; (parent * 2 + 1) < heap.size; parent = child)
	{
		child = parent * 2 + 1;

		//取两个子节点的最大的节点
		if (((child + 1) < heap.size) && (heap.arr[child] < heap.arr[child + 1]))
		{
			child++;
		}

		//判断最大的节点是否大于当前的父节点
		if (cur >= heap.arr[child])//不大于,则不需要调整,跳出循环
		{
			break;
		}
		else//大于当前的父节点,进行交换,然后从子节点位置继续向下调整
		{
			heap.arr[parent] = heap.arr[child];
			heap.arr[child] = cur;
		}
	}

}

插入新元素

将数字99 插入到上面大顶堆中的过程如下:
1.原始的堆,如图a

对应的数组:{95, 93, 87, 92, 86, 82}
  1. 将新进的元素插入到大顶堆的尾部,如下图b所示:

对应的数组:{95, 93, 87, 92, 86, 82, 99}
  1. 此时最大堆已经被破坏,需要重新调整,因加入的节点比父节点大,则新节点跟父节点调换即可,如图c所示;调整后,新节点如果比新的父节点小,则已经调整到位,如果比新的父节点大,则需要和父节点重新进行交换,如图d,至此,最大堆调整完成。

代码实现:

#include <stdio.h> 
#include <stdlib.h>
#include <string.h>
#define DEFAULT_CAPCITY 128

typedef struct _Heap 
{
    int *arr; //存储堆元素的数组
    int size; //当前已存储的元素个数
    int capacity; //当前存储的容量
}Heap;

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

/*初始化堆*/
bool initHeap(Heap &heap, int *orginal, int size)
{
    int capacity = DEFAULT_CAPCITY>size ? DEFAULT_CAPCITY : size;
    heap.arr = new int[capacity];
    if (!heap.arr)   return false;
    heap.capacity = capacity;
    heap.size = 0;
   
    //如果存在原始数据则构建堆
    if(size > 0)
    {
        /*方式一: 直接调整所有元素
        memcpy(heap.arr, orginal, size*sizeof(int));
        heap.size = size;
        //建堆
        buildHeap(heap);
        */
        
        //方式二: 一次插入一个
        for(int i=0; i<size; i++)
        {
            insert(heap, orginal[i]);
        }
    }
    return true;
}

/* 从最后一个父节点(size/2-1 的位置)逐个往前调整所有父节点(直到根节点),确保每一个父节点都是一个最大堆,最后整体上形成一个最大堆*/
void buildHeap(Heap &heap) 
{
    int i;
    for (i = heap.size / 2 - 1; i >= 0; i--) 
    {
        adjustDown(heap, i);
    }
}

/*将当前的节点和子节点调整成最大堆*/
void adjustDown(Heap &heap, int index)
{
    int cur = heap.arr[index];//当前待调整的节点
    int parent, child;
    /*
    判断否存在大于当前节点子节点,如果不存在,则堆本身是平衡的,不需要调整;
    如果存在,则将最大的子节点与之交换,交换后,如果这个子节点还有子节
    点,则要继续按照同样的步骤对这个子节点进行调整
    */
    for (parent = index; (parent * 2 + 1)<heap.size; parent = child) 
    {
        child = parent * 2 + 1;
        
        //取两个子节点中的最大的节点
        if (((child + 1)<heap.size) && (heap.arr[child]<heap.arr[child +1]))
        {
            child++;
        }
        
        //判断最大的节点是否大于当前的父节点
        if (cur >= heap.arr[child])
        {
            //不大于,则不需要调整,跳出循环
            break;
        }
        else//大于当前的父节点,进行交换,然后从子节点位置继续向下调整
        {
            heap.arr[parent] = heap.arr[child];
            heap.arr[child] = cur;
        }
    }
}

/*将当前的节点和父节点调整成最大堆*/
void adjustUp(Heap &heap, int index)
{
    if(index<0 || index >= heap.size)//大于堆的最大值直接return
    {
        return;
    }
    
    while(index>0)
    {
        int temp = heap.arr[index];
        int parent = (index - 1) / 2;
        if(parent >= 0)//如果索引没有出界就执行想要的操作
        {
            if(temp > heap.arr[parent])
            {
                heap.arr[index] = heap.arr[parent];
                heap.arr[parent] = temp;
                index = parent;
            }
            else//如果已经比父亲小直接结束循环
            {
                break;
            }
        }
        else//越界结束循环
        {
            break;
        }
    }
}

/*最大堆尾部插入节点,同时保证最大堆的特性*/
bool insert(Heap &heap, int value)
{
    if (heap.size == heap.capacity)
    {
        fprintf(stderr, "栈空间耗尽!\n");
        return false;
    }
   
    int index = heap.size;
    heap.arr[heap.size++] = value;
    adjustUp(heap, index);
    return true;
}

int main(void)
{
    Heap hp;
    int origVals[] = { 1, 2, 3, 87, 93, 82, 92, 86, 95 };
    int i = 0;
    if(!initHeap(hp, origVals, sizeof(origVals)/sizeof(origVals[0])))
    {
        fprintf(stderr, "初始化堆失败!\n");
        exit(-1);
    }
    for (i = 0; i<hp.size; i++) 
    {
        printf("the %dth element:%d\n", i, hp.arr[i]);
    }
    insert(hp, 99);
    printf("在堆中插入新的元素99,插入结果:\n");
    for (i = 0; i<hp.size; i++)
    {
        printf("the %dth element:%d\n", i, hp.arr[i]);
    }
    system("pause");
    return 0;
}

参考资料来源:

奇牛学院

posted @ 2023-06-12 13:39  CodeMagicianT  阅读(22)  评论(0编辑  收藏  举报