堆排序的实现 C语言
/*heap sort*/
/*交换数组中的两个元素*/
void swap(int arr[], int idx1, int idx2)
{
int tmp = arr[idx1];
arr[idx1] = arr[idx2];
arr[idx2] = tmp;
return;
}
/*堆化,根据定义调整为大顶堆*/
void max_heapify(int arr[], int index, int num)
{
int largest_idx = 0; /*保存最大值节点的索引*/
/*根据堆的性质,找出其左右节点的索引*/
int left = 2 * index + 1;
int right = 2 * index + 2;
/*默认当前节点(left和right的父节点)为最大值*/
largest_idx = index;
/*如果有左节点,且左节点值更大,则更新最大值的索引*/
if((left < num) && (arr[left] > arr[largest_idx]))
{
largest_idx = left;
}
/*如果有右节点,且右节点值更大,则更新最大值索引*/
if((right < num) && (arr[right] > arr[largest_idx]))
{
largest_idx = right;
}
/*如果最大值不是当前(index)非叶节点的值,则将当前节点和最大的子节点值交换*/
if(largest_idx != index)
{
swap(arr, index, largest_idx);
/*由于交换后子节点值变了,如果该子节点也有自己的子节点,则需再次调整*/
max_heapify(arr, largest_idx, num);
}
return;
}
/*创建大顶堆:把待排序序列,变成一个大顶堆结构的数组*/
void build_max_heap(int arr[], int len)
{
int i = 0;
int last_not_leaf_node = len / 2 - 1; /*最后一个非叶节点*/
/*从下往上调整*/
for(i = last_not_leaf_node; i >= 0; i--)
{
max_heapify(arr, i, len);
}
return;
}
/*堆排序函数*/
void heap_sort(int arr[], int size)
{
int i = 0;
if((NULL == arr) || (size <= 0))
{
return;
}
/*根据堆定义创建大顶堆:其实是把待排列的序列变成一个大顶堆结构的数组*/
build_max_heap(arr, size);
/*交换堆顶和末尾节点,然后重置大顶堆*/
for(i = size-1; i > 0; i--)
{
swap(arr, 0, i);
/*末尾节点已是最大,剔除*/
--size;
/*将剩余节点继续调整为大顶堆*/
max_heapify(arr, 0, size);
}
return;
}