编辑代码

#include <stdio.h>
// 交换数组中的两个元素
void swap(int* a, int* b) {
    int temp = *a;
    *a = *b;
    *b = temp;
}
// 调整堆
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) {
        swap(&arr[i], &arr[largest]);
        heapify(arr, n, largest);
    }
}
// 堆排序
void heapSort(int arr[], int n) {
    // 建堆
    for (int i = n / 2 - 1; i >= 0; i--)
        heapify(arr, n, i);
    // 一个个取出堆顶元素，进行排序
    for (int i = n - 1; i > 0; i--) {
        swap(&arr[0], &arr[i]);  // 将当前最大元素移动到数组末尾
        heapify(arr, i, 0);      // 重新调整堆
    }
}
// 打印数组
void printArray(int arr[], int size) {
    for (int i = 0; i < size; i++)
        printf("%d ", arr[i]);
    printf("\n");
}
int main() {
    int arr[] = {12, 11, 13, 5, 6, 7};
    int n = sizeof(arr) / sizeof(arr[0]);
    printf("Unsorted array: \n");
    printArray(arr, n);
    heapSort(arr, n);
    printf("Sorted array: \n");
    printArray(arr, n);
    return 0;
}