个人主页:仍有未知等待探索-CSDN博客
专题分栏:数据结构_仍有未知等待探索的博客-CSDN博客
欢迎大家来指教!
一、前言
今天要介绍的是堆排序。
首先什么是堆?简而言之,堆就是二叉树的数组形式,用数组来存储二叉树。
这个堆和C语言中讲的堆区是不同的两个概念,不要混淆。
二、堆排序
堆排序的核心就是构建一个特殊的二叉树,这个二叉树的特性是:其父节点大于等于(小于等于)其左右孩子结点。
故,最终创建的二叉树的根节点会是该二叉树的最大值(最小值)。
那怎么才能让整个数组中的数据有序呢?那我们就让数组中第一个数和最后一个数进行交换,然后以第一个数和倒数第二个数为端点,继续进行构建堆,然后选出次最小值。以此类推~~~
我以大根堆为例(从数组下标为1的地方开始存储):
1、堆的数据类型
typedef int HpDataType;
typedef struct Heap
{
HpDataType* a;//进行存储堆中的数据
int size;//该堆的有效数据个数
int capacity;//该堆中的容量
}Hp;
2、堆的初始化
//初始化
void Heapinit(Hp* php)
{
assert(php);//断言,如果php为空的话,报错
php -> a = NULL;
php -> size = 0;
php -> capacity = 0;
}
3、堆的销毁
//堆的销毁
void HeapDestory(Hp* php)
{
assert(php);
free(php -> a);//释放掉开辟的空间
php -> a = NULL;//防止php -> a为野指针
php -> size = 0;
php -> capacity = 0;
}
4、堆的创建
// 堆的创建
// k为要插入到堆里面的元素
void Heap_push(Hp* php, int k)
{
assert(php); //断言
if (php -> size == php -> capacity)// 如果容量不够,扩容
{
// 如果容量为0,则给个初始值,否则就二倍扩容
int size = php -> capacity == 0 ? 4 : php -> capacity * 2;
Hp* tmp = (Hp* ) realloc(php -> a, (size + 1) * sizeof(int));
if (tmp == NULL)
{
perror("realloc failed");
exit(-1);// 扩容失败则结束程序
}
php -> a = tmp;
php -> capacity = size;
}
// 插入到堆的最后
php -> a[++ php -> size] = k;
// 向上调整,如果比父节点要大就进行交换
up(php -> a, php -> size);
}
5、向上调整
// 向上调整
// idx为要调整元素的下标
void up(int* a, int idx)
{
assert(a);
int child = idx;// 该结点下标
int parent = child / 2;// 该结点的父节点,我是从下标为1开始存的
while (child != 1)
{
if (a[parent] < a[child])
{
Swap(&a[parent], &a[child]);
}
else
{
// 不需要调整,就退出
break;
}
// 继续向上找
child = child / 2;
parent = parent / 2;
}
}
6、删除堆的最后一个元素
该操作的主要作用是将该堆中最大的元素进行归位,并且进行调整除了最后一个元素的二叉树的顺序,使其仍为一个堆。
// 进行删除最后一个元素
void Heappop(Hp* php)
{
assert(php);
assert(php -> size > 0);
Swap(&php -> a[1], &php -> a[php -> size]);
php -> size --;
int parent = 1;
int child = 2 * parent;
while (child <= php -> size)
{
// 假设左节点最大,如果右结点比左节点大,就child存右节点下标
if (child + 1 <= php -> size && php -> a[child + 1] > php -> a[child])
{
child += 1;
}
// 判断孩子结点和父节点的值
if (php -> a[parent] < php -> a[child])
{
Swap(&php -> a[parent], &php -> a[child]);
}
else
{
// 不需要则退出
break;
}
parent = child;
child = 2 * parent;
}
}
7、取出堆顶元素
// 取出堆顶元素
HpDataType Heaptop(Hp* php)
{
assert(php);
assert(php -> size > 0);
return php -> a[1];
}
三、总代码及其运行结果
#define _CRT_SECURE_NO_WARNINGS 1
// 大根堆
#include <stdio.h>
#include <assert.h>
#include <stdlib.h>
typedef int HpDataType;
typedef struct Heap
{
HpDataType* a;// 进行存储堆中的数据
int size;// 该堆的有效数据个数
int capacity;// 该堆中的容量
}Hp;
void Heapinit(Hp* php);// 初始化
void HeapDestory(Hp* php);// 堆的销毁
void Heap_push(Hp* php, int k);// 堆的创建
void Swap(int* a, int* b);// 交换
void up(int* a, int idx);// 向上调整
HpDataType Heaptop(Hp* php);// 取堆顶元素
void Heappop(Hp* php);// 删除堆的最后一个元素
int main()
{
int a[] = {1, 2, 3, 4, 5, 6, 7};
Hp hp;
Heapinit(&hp);
int len = sizeof(a) / sizeof(a[0]);
for (int i = 0; i < len; i++ )
{
Heap_push(&hp, a[i]);
}
printf("初始堆:\n");
for (int i = 1; i <= len; i ++ )
{
printf("%d ", hp.a[i]);
}
printf("\n有序序列:\n");
while (hp.size > 0)
{
printf("%d ", Heaptop(&hp));
Heappop(&hp);
}
HeapDestory(&hp);
return 0;
}
// 初始化
void Heapinit(Hp* php)
{
assert(php);// 断言,如果php为空的话,报错
php -> a = NULL;
php -> size = 0;
php -> capacity = 0;
}
// 堆的销毁
void HeapDestory(Hp* php)
{
assert(php);
free(php -> a);// 释放掉开辟的空间
php -> a = NULL;// 防止php -> a为野指针
php -> size = 0;
php -> capacity = 0;
}
void Swap(int* a, int* b)
{
int tmp = *a;
*a = *b;
*b = tmp;
}
// 向上调整
// idx为要调整元素的下标
void up(int* a, int idx)
{
assert(a);
int child = idx;// 该结点下标
int parent = child / 2;// 该结点的父节点,我是从下标为1开始存的
while (child != 1)
{
if (a[parent] < a[child])
{
Swap(&a[parent], &a[child]);
}
else
{
// 不需要调整,就退出
break;
}
// 继续向上找
child = child / 2;
parent = parent / 2;
}
}
// 堆的创建
// k为要插入到堆里面的元素
void Heap_push(Hp* php, int k)
{
assert(php); //断言
if (php -> size == php -> capacity)// 如果容量不够,扩容
{
// 如果容量为0,则给个初始值,否则就二倍扩容
int size = php -> capacity == 0 ? 4 : php -> capacity * 2;
Hp* tmp = (Hp* ) realloc(php -> a, (size + 1) * sizeof(int));
if (tmp == NULL)
{
perror("realloc failed");
exit(-1);// 扩容失败则结束程序
}
php -> a = tmp;
php -> capacity = size;
}
// 插入到堆的最后
php -> a[++ php -> size] = k;
// 向上调整,如果比父节点要大就进行交换
up(php -> a, php -> size);
}
// 取出堆顶元素
HpDataType Heaptop(Hp* php)
{
assert(php);
assert(php -> size > 0);
return php -> a[1];
}
// 进行删除最后一个元素
void Heappop(Hp* php)
{
assert(php);
assert(php -> size > 0);
Swap(&php -> a[1], &php -> a[php -> size]);
php -> size --;
int parent = 1;
int child = 2 * parent;
while (child <= php -> size)
{
// 假设左节点最大,如果右结点比左节点大,就child存右节点下标
if (child + 1 <= php -> size && php -> a[child + 1] > php -> a[child])
{
child += 1;
}
// 判断孩子结点和父节点的值
if (php -> a[parent] < php -> a[child])
{
Swap(&php -> a[parent], &php -> a[child]);
}
else
{
// 不需要则退出
break;
}
parent = child;
child = 2 * parent;
}
}
谢谢大家!