本文中的代码适用于对数组元素进行排序(以整型数据为例)。
关于堆排序的具体思路可以参考《算法》书中描述 堆排序 。
自定义的方法
在排序方法之前,定义了五个不同的方法,以便于对数组元素进行比较、交换和覆盖。为了适用于不同的数据类型,这里使用到了 void 指针以及指针类型的转换,也就是 C 语言中范型的概念。如果需要对不同元素类型的数组进行排序,只修改 LessArr() 方法即可。
// 比较数组中两个元素大小的方法
int LessArr(void *arr, int i, int j, int size)
{
return *(int*) (arr + (i - 1) * size) < *(int*) (arr + (j - 1) * size);
}
// 交换数组中两个元素的方法
void Exch(void *arr, int i, int j, int size)
{
void * temp = malloc(size);
memcpy(temp, arr + (i - 1) * size, size);
memcpy(arr + (i - 1) * size, arr + (j - 1) * size, size);
memcpy(arr + (j - 1) * size, temp, size);
free(temp);
}
// 实现数组中元素的覆盖
void Cover(void * arr, int i, int j, int size)
{
memcpy(arr + (i - 1) * size, arr + (j - 1) * size, size);
}
// 数组中元素的下沉方法
void Sink(void *arr, int k, int n, int size)
{
while (2 * k <= n)
{
int j = 2 * k;
if (j < n && LessArr(arr, j, j + 1, size)) ++j;
if (!LessArr(arr, k, j, size)) break;
Exch(arr, k, j, size);
k = j;
}
}
// 数组中元素的上浮方法
void Swim(void *arr, int k,