编辑代码

#include <stdio.h>

void heapify(int arr[], int n, int i) {
    int largest = i;
    int left_child = 2 * i + 1;
    int right_child = 2 * i + 2;

    // 比较根节点与左子节点
    if (left_child < n && arr[left_child] > arr[largest]) {
        largest = left_child;
    }

    // 比较根节点与右子节点
    if (right_child < n && arr[right_child] > arr[largest]) {
        largest = right_child;
    }

    // 如果最大元素的索引不是根节点，交换根节点和最大元素
    if (largest != i) {
        int temp = arr[i];
        arr[i] = arr[largest];
        arr[largest] = temp;
        // 递归调整交换后的子堆
        heapify(arr, n, largest);
    }
}

void build_heap(int arr[], int n) {
    // 从最后一个非叶子节点开始，逐步向前构建最大堆
    for (int i = n / 2 - 1; i >= 0; i--) {
        heapify(arr, n, i);
    }
}

void heap_sort(int arr[], int n) {
    // 构建最大堆
    build_heap(arr, n);

    // 逐步将最大元素交换到数组末尾，并缩小堆的范围
    for (int i = n - 1; i > 0; i--) {
        // 交换堆顶元素和数组末尾元素
        int temp = arr[0];
        arr[0] = arr[i];
        arr[i] = temp;

        // 缩小堆的范围并重新调整堆
        heapify(arr, i, 0);
    }
}

// 示例
int main() {
    int arr[] = {20, 11, 15, 22, 6, 7};
    int n = sizeof(arr) / sizeof(arr[0]);

    // 排序数组
    heap_sort(arr, n);

    // 打印排序后的数组
    printf("排序后的数组: ");
    for (int i = 0; i < n; i++) {
        printf("%d ", arr[i]);
    }
    printf("\n");

    return 0;
}