算法-堆排序

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

相关推荐
我命由我1234532 分钟前
人脸识别 - 人脸识别选帧
java·人工智能·python·算法·安全·java-ee·人脸识别
时针滴滴答啊34 分钟前
最大子数组和
算法·leetcode·职场和发展
五月底_1 小时前
LIS算法
python·算法·最长子序列
学学酱快乐2 小时前
2026年PMP考试生命周期选择决策树精讲:从需求判断到混合型统一框架的应试全攻略
算法·决策树·机器学习·pmp新版考试大纲·pmp新题型·pmp培训哪家好·pmp培训机构推荐
傅科摆 _ py2 小时前
动态规划复习
算法·动态规划
Wang's Blog2 小时前
PostgreSQL笔记49:向量检索核心算法、索引调优与过滤策略深度解析
笔记·算法·postgresql
疯狂打码的少年2 小时前
【数据结构】交换类排序:冒泡与快速排序
数据结构·笔记·算法·排序算法
Nil2083 小时前
leetcode 108有序数组转换为二叉搜索树
数据结构·算法·leetcode
高频因子挖掘机3 小时前
QuantDash 成交量单位统一实战:从“手”到“股”的跨市场量化数据清洗全流程
后端·算法·github
hn小菜鸡3 小时前
LeetCode 763、划分字母区间
数据结构·算法·leetcode