堆排序 (Heap Sort)
像"选拔赛"一样,每次把数组构造成一个"大顶堆"(保证最顶端是最大值)。然后把冠军(最大值)和队尾的人交换,让他提前"退休"到末尾。接着把剩下的队伍重新调整回大顶堆,再选出新的冠军换到倒数第二个位置。不断重复,队伍就从小到大排好序了。
核心代码
c
#include <stdio.h>
// 堆化函数(向下调整):维护大顶堆的性质
// n 为当前堆的有效长度,i 为当前需要调整的父节点下标
void heapify(int arr[], int n, int i) {
int largest = i; // 假设当前父节点是最大的
int left = 2 * i + 1; // 左孩子下标
int right = 2 * i + 2;// 右孩子下标
// 如果左孩子存在且大于当前最大值,更新最大值下标
if (left < n && arr[left] > arr[largest]) {
largest = left;
}
// 如果右孩子存在且大于当前最大值,更新最大值下标
if (right < n && arr[right] > arr[largest]) {
largest = right;
}
// 如果最大值不是父节点本身,说明需要交换,并继续向下调整
if (largest != i) {
int temp = arr[i];
arr[i] = arr[largest];
arr[largest] = temp;
heapify(arr, n, largest); // 递归调整受影响的子树
}
}
// 堆排序函数
void heapSort(int arr[], int n) {
// 1. 建堆:从最后一个非叶子节点开始,自底向上调整为大顶堆
for (int i = n / 2 - 1; i >= 0; i--) {
heapify(arr, n, i);
}
// 2. 排序:每次将堆顶(最大值)与末尾元素交换,并缩小堆的范围
for (int i = n - 1; i > 0; i--) {
int temp = arr[0];
arr[0] = arr[i];
arr[i] = temp;
heapify(arr, i, 0); // 对剩下的 i 个元素重新堆化
}
}
// 打印数组的辅助函数
void printArray(int arr[], int n) {
for (int i = 0; i < n; i++) {
printf("%d ", arr[i]);
}
printf("\n");
}
int main() {
int arr[] = {64, 34, 25, 12, 22};
int n = sizeof(arr) / sizeof(arr[0]); // 计算数组长度
printf("排序前:");
printArray(arr, n);
heapSort(arr, n); // 调用堆排序
printf("排序后:");
printArray(arr, n);
return 0;
}
示例演示
{64, 34, 25, 12, 22}
第 1 步:构建初始大顶堆
目标:将无序数组调整为父节点大于等于子节点的大顶堆。
最后一个非叶子节点下标为 5/2 - 1 = 1(元素 34)。比较 34 和它的孩子 12、22,无需交换。
处理下标 0(元素 64)。比较 64 和它的孩子 34、25,64 最大,无需交换。
建堆完成,堆顶 64 为全局最大值。
当前数组:64, 34, 25, 12, 22
第 2 轮:选出最大值 64目标:将堆顶 64 与末尾元素 22 交换,并对前 4 个元素重新堆化。
交换 64 和 22,数组变为:22, 34, 25, 12, \| 64
对 22 进行向下调整:22 的左孩子是 34,右孩子是 25,34 最大,交换 22 和 34。
第 2 轮结束:最大值 64 归位。
当前数组:34, 22, 25, 12, \| 64
第 3 轮:选出第二大值 34目标:将堆顶 34 与倒数第二个元素 12 交换,并对前 3 个元素重新堆化。
交换 34 和 12,数组变为:12, 22, 25, \| 34, 64
对 12 进行向下调整:12 的孩子是 22 和 25,25 最大,交换 12 和 25。
第 3 轮结束:第二大值 34 归位。
当前数组:25, 22, 12, \| 34, 64
第 4 轮:选出第三大值 25目标:将堆顶 25 与倒数第三个元素 12 交换,并对前 2 个元素重新堆化。
交换 25 和 12,数组变为:12, 22, \| 25, 34, 64
对 12 进行向下调整:12 的左孩子是 22,22 > 12,交换 12 和 22。
第 4 轮结束:第三大值 25 归位。
当前数组:22, 12, \| 25, 34, 64
第 5 轮:选出第四大值 22目标:将堆顶 22 与倒数第四个元素 12 交换,堆只剩 1 个元素,无需调整。
交换 22 和 12,数组变为:12, \| 22, 25, 34, 64
排序完成。
最终结果:12, 22, 25, 34, 64
复杂度
时间复杂度是 \(O(nlogn)\) (建堆耗时 \(O(n)\),每次调整堆耗时 \(O(logn)\),共调整 n−1 次)
空间复杂度是 \(O(1)\)(原地排序,仅需常数级的额外空间用于交换变量)