首先,我们来了解什么是堆以及它有什么特点。
堆是一种特殊的完全二叉树,每个节点都有一个值,且每个节点的值都大于或小于其子节点的值。根据这个特性,堆可以分为最大堆和最小堆两种。最大堆的每个节点都大于或等于其子节点,而最小堆的每个节点都小于或等于其子节点。示意图如下:
根据上述图片可以看出,最大堆的根节点的值是所有节点值中的最大值,最小堆的根节点的值是所有节点值中的最小值。因此,我们可以通过将一堆数据构成堆,再根据这个特点来获取最大值(最小值)或者排序。
下面介绍如何将一堆数据构成堆:
数组构成堆过程中的特点:parent(父节点的下标)=(child(子节点的下标)-1)/2;
child=(parent*2)+1;
首先将这些数据(5,10,9,1,6,12 )构成一个数组int a[] = { 5,10,9,1,6,12 }。
然后依次将数组的每个元素插入完全二叉树,再进行向下调整,使该二叉树符合堆的特点。
向下调整的过程:
向二叉树插入数据为6的节点,比较插入节点的值(child)与其父节点的值(parent)的大小,如果插入节点的值大,则不进行操作;如果插入节点的值小,则插入节点的值与其父节点的值互换,插入节点下标(child)=其父节点下标(parent),父节点下标(parent)=parent*2+1,然后再比较新子节点的值(插入节点的值)与新父节点的值,进行重复操作。操作示意图:
堆排序过程:
交换根节点(数组第一个数据)和最后节点(数组末尾数据)的值,获取交换后最后节点的值,让
数组总数减1,再进行向下调整。重复上述步骤就可以获得一组有序序列,也就是堆排序过程。
具体代码实现如下:
#include<stdio.h>
#include<assert.h>
//交换两个数
void Swap(int* a, int* b) {
int c = *a;
*a = *b;
*b = c;
}
//向下调整
void Adjustdown(int* a, int size,int parent) {
assert(a);
int child = (parent*2)+1;
while (child < size) {
if (child+1<size&&a[child + 1] > a[child]) {
++child;
}
if (a[child] > a[parent])
{
Swap(&a[parent], &a[child]);
parent = child;
child = (parent * 2) + 1;
}
else {
break;
}
}
}
void HeapSort(int* a, int n) {
for (int i =(n-2)/2; i >=0; i--) {
Adjustdown(a, n, i);//建大堆
}
int end = n - 1;
while (end > 0) {
Swap(&a[0], &a[end]);
Adjustdown(a, end, 0);//从根向下调整
--end;
}
}
int main()
{
int a[] = { 5,10,9,1,6,12 };
HeapSort(a, sizeof(a) / sizeof(int));
for (int i = 0; i < sizeof(a) / sizeof(int); i++) {
printf("%d ", a[i]);
}
return 0;
}
文章如有错误,望大家指正。