堆排序 代码+注释（C语言）

#include <stdio.h>

// 打印数组元素
void printArray(int arr[], int n) {
    for (int i = 0; i < n; i++)
        printf("%d ", arr[i]);
    printf("\n");
}

// 交换数组元素
void swap(int arr[], int index1, int index2)
{
    int temp = arr[index1];
    arr[index1] = arr[index2];
    arr[index2] = temp;
}

// 维护节点
/*
    len 元素个数、节点数量
    i 需要维护的节点下标
*/
void heapify(int arr[], int len, int i)
{
    int largest = i;            // 当前子树的根节点下标，假设其值最大
    int lson = i * 2 + 1;       // 左子节点下标
    int rson = i * 2 + 2;       // 右子节点下标

    // 如果左子节点存在且大于当前根节点，更新最大节点下标
    if (lson < len && arr[largest] < arr[lson])
    {
        largest = lson;
    }

    // 如果右子节点存在且大于当前根节点，更新最大节点下标
    if (rson < len && arr[largest] < arr[rson])
    {
        largest = rson;
    }

    // 如果最大节点不是根节点，交换最大节点和根节点的值，并递归调整子树
    if (largest != i)
    {
        swap(arr, largest, i);
        heapify(arr, len, largest);  
    }
}

void heapSort(int arr[], int len)
{
    // 创建堆，从最后一个非叶节点开始往前遍历，依次维护堆性质
    for (int i = len / 2 - 1; i >= 0; i--)
    {
        heapify(arr, len, i);
    }

    // 堆排序
    for (int i = len - 1; i > 0; i--)
    {
        swap(arr, i, 0);          // 将堆顶元素（最大值）移到末尾
        heapify(arr, i, 0);       // 调整堆，将剩余部分重新构建成最大堆
    }
}

int main() {
    int arr[] = { 12, 11, 13, 5, 6, 7 };
    int n = sizeof(arr) / sizeof(arr[0]);

    printf("原数组：\n");
    printArray(arr, n);

    heapSort(arr, n);

    printf("排序后的数组：\n");
    printArray(arr, n);

    return 0;
}