实现堆的各种基本运算的算法(数据结构)

以小堆为例,大堆就举一反三了。

堆的物理结构就是普通的数组,但是逻辑结构看成了一颗完全二叉树。

小堆,就是树的每一个父节点都小于他的孩子节点。如图中第一排的a与b。大堆,就是树的每一个父节点都大于他的孩子节点。如图中第二排的a与b。我们要实现的功能就是将一个数组排成符合要求的序列。

创建小堆

cpp 复制代码
typedef int HeapDataType;

typedef struct Heap//小堆
{
	HeapDataType* a;
	int size;
	int capacity;
}HP;

堆初始化

cpp 复制代码
//初始化堆
void HeapInit(HP* php)
{
	assert(php);
	php->a = NULL;
	php->size = php->capacity = 0;
}

向堆增加数据

cpp 复制代码
//Push数据
void HeapPush(HP* php, HeapDataType x)
{
	assert(php);
	if (php->size == php->capacity)
	{
		int newcapacity = php->capacity == 0 ? 4 : php->capacity * 2;
		HeapDataType* temp = (HeapDataType*)realloc(php->a, sizeof(HeapDataType) * newcapacity);
		if (temp == NULL)
		{
			printf("realloc fail\n");
			exit(-1);
		}
		php->capacity = newcapacity;
		php->a = temp;
	}
	php->a[php->size++] = x;
	HeapAdjustUp(php->a, php->size-1);
}

删除堆顶数据

每一次删除堆顶元素,都会再次形成一个新的堆,新堆的堆顶元素就是数组a中次

小的值,再次删除,再形成新堆,堆顶就变成次次小的元素了。以此类推。

cpp 复制代码
//Pop数据
void HeapPop(HP* php)
{
	assert(php);
	assert(php->size > 0);
	Swap(&php->a[0], &php->a[php->size - 1]);
	php->size--;
	HeapAdjustDown(php->a, php->size, 0);
}

获得堆顶数据

cpp 复制代码
//获得堆顶数据
HeapDataType HeapTop(HP* php)
{
	assert(php);
	return php->a[0];
}

获得堆的有效大小

cpp 复制代码
//获得堆的有效大小
size_t HeapSize(HP* php)
{
	return php->size;
}

判断堆是否为空

cpp 复制代码
//判断堆是否为空
bool HeapEmpty(HP* php)
{
	assert(php);
	return php->size == 0 ? 1 : 0;
}

交换数据

cpp 复制代码
//交换数据
void Swap(HeapDataType* p1,HeapDataType* p2)
{
	HeapDataType p3 = *p1;
	*p1 = *p2;
	*p2 = p3;
}

向上调整数组形成堆

cpp 复制代码
//向上调整数组形成堆
void HeapAdjustUp(HeapDataType* a, int child)
{
	int parent = (child - 1) / 2;
	while (child > 0)
	{
		if (a[parent] > a[child])
		{
			Swap(&a[parent], &a[child]);
			child = parent;
			parent = (child - 1) / 2;
		}
		else
		{
			break;
		}
	}
}

向下调整数组形成堆

cpp 复制代码
//向下调整数组形成堆
void HeapAdjustDown(int* a, int size, int parent)
{
	int child = parent * 2 + 1;
	while (child < size)
	{
		if (child + 1< size && a[child + 1] < a[child])
		{
			child++;
		}
		if (a[parent] > a[child])
		{
			Swap(&a[parent], &a[child]);
			parent = child;
			child = parent * 2 + 1;
		}
		else
		{
			break;
		}
	}
}

销毁堆

cpp 复制代码
//销毁堆
void HeapDestory(HP* php)
{
	assert(php);
	free(php->a);
	php->a = NULL;
	php->size = php->capacity = 0;
}

所需头文件

cpp 复制代码
#pragma once
#include "stdio.h"
#include "stdlib.h"
#include "string.h"
#include "stdbool.h"
#include "assert.h"

main函数测试

cpp 复制代码
#include "Heap.h"

int main()
{
	int a[] = { 4,6,2,1,5,8,2,9};
	HP hp;//创建堆
	HeapInit(&hp);//初始化堆
    
   //将数组a内数据Push到堆中
	for (int i = 0; i < sizeof(a) / sizeof(int); ++i)
	{
		HeapPush(&hp, a[i]);
	}

	
	int k = 3;
	while (k--)
	{
		printf("%d\n", HeapTop(&hp));
		HeapPop(&hp);//每一次删除堆顶元素,都会再次形成一个新的堆,新堆的堆顶元素就是数组a中次
                    //小的值,再次删除,再形成新堆,堆顶就变成次次小的元素了。以此类推。
                    //所以打出来的会是升序的
	}

	while (!HeapEmpty(&hp))
	{
		printf("%d\n", HeapTop(&hp));
		HeapPop(&hp);
	}
	printf("\n");

	return 0;
}

结果

相关推荐
l1t2 分钟前
利用小米mimo为精确覆盖矩形问题C程序添加打乱函数求出更大的解
c语言·开发语言·javascript·人工智能·算法
亭上秋和景清4 分钟前
strlen;strcpy ;strcat
算法
_OP_CHEN4 分钟前
【算法基础篇】(三十五)图论基础之最小生成树:从原理到实战,彻底吃透 Prim 与 Kruskal 算法
算法·蓝桥杯·图论·最小生成树·kruskal算法·prim算法·acm/icpc
LYFlied10 分钟前
【算法解题模板】-【回溯】----“试错式”问题解决利器
前端·数据结构·算法·leetcode·面试·职场和发展
拾忆,想起13 分钟前
设计模式:软件开发的可复用武功秘籍
开发语言·python·算法·微服务·设计模式·性能优化·服务发现
lxh011319 分钟前
最长有效括号
数据结构·算法
橙子牛奶糖22 分钟前
Science | 本周最新文献速递
算法·gwas·生物信息学·单细胞测序
fei_sun24 分钟前
【总结】【数据结构】【OS】【计组】【计网】
数据结构
皮卡蛋炒饭.25 分钟前
背包问题Ⅱ与二分问题
算法
Code Slacker28 分钟前
LeetCode Hot100 —— 普通数组(面试纯背版)(五)
数据结构·c++·算法·leetcode·面试