算法-堆排序

堆排序 (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)\)(原地排序,仅需常数级的额外空间用于交换变量)

相关推荐
星马梦缘2 小时前
算法设计 期末复习笔记
算法·动态规划·二分图·最小生成树·线性规划·最大流
鱼儿也有烦恼2 小时前
题型归纳与重要模版
算法
菜鸟不爱学3 小时前
第P4周:猴痘病识别
算法
QN1幻化引擎3 小时前
我写了一个「有八层意识结构」的 Python 认知引擎,它没有用任何 LLM
人工智能·算法·架构
j7~4 小时前
【数据结构初阶】顺序表增删查改代码实现--详解
数据结构·顺序表·动态顺序表·静态顺序表
枕星而眠4 小时前
【数据结构】红黑树入门指南
运维·数据结构·c++·后端
Keven_115 小时前
算法札记:如何卡SPFA
算法·spfa
imuliuliang5 小时前
关于并发哈希表的锁分段机制与冲突减少策略7
算法
可编程芯片开发5 小时前
基于电压电流双闭环控制的三相整流器系统simulink建模与仿真
算法