C算法——排序 堆排序算法

源码+注释

//
// Created by Lenovo on 2022-06-14-下午 7:42.
// 作者：小象
// 版本：1.0
//

#include <stdio.h>

#define LENGTH(arr) (sizeof(arr)/sizeof((arr)[0])) // 计算数组长度

void heapSort(int arr[], int length);

void adjustHeap(int arr[], int i, int length);

int main() {

    int arr[] = {1, 80, 2, 57, 13};
    printf("排序前 arr = ");
    for (int i = 0; i < LENGTH(arr); i++) {
        printf("%d ", arr[i]);
    }
    printf("\n----------------------------\n");

    heapSort(arr, LENGTH(arr));

    printf("排序后 arr = ");
    for (int i = 0; i < LENGTH(arr); i++) {
        printf("%d ", arr[i]);
    }
    printf("\n----------------------------\n");

    return 0;
}

void heapSort(int arr[], int length) {
    int temp;
    // 将无序序列构建成一个堆，根据升序降序需求选择大顶堆或小顶堆
    for (int i = length / 2 - 1; i >= 0; i--) {
        adjustHeap(arr, i, length);
    }
    for (int i = length - 1; i > 0; i--) {
        // 将堆顶元素与末尾元素交换，将最大元素"沉"到数组末端
        temp = arr[i];
        arr[i] = arr[0];
        arr[0] = temp;
        // 重新调整结构，使其满足堆定义，然后继续交换堆顶元素与当前末尾元素，反复执行调整+交换步骤，直到整个序列有序
        adjustHeap(arr, 0, i); // 从最顶端 下标0 开始重新调整为堆
    }
}

void adjustHeap(int arr[], int i, int length) {
    // 取出当前元素的值，保存为临时变量
    int temp = arr[i];
    // k=i*2+1：k是i结点的左子结点
    for (int k = i * 2 + 1; k < length; k = k * 2 + 1) {
        // 在保证有右子结点的前提下，比较左子结点和右子结点的值的大小
        if (k + 1 < length && arr[k] < arr[k + 1]) {
            k++; // 如果右子结点大于左子结点，则把 左子结点 赋值为 右子结点
        }
        // 如果子结点大于父结点
        if (arr[k] > temp) {
            arr[i] = arr[k]; // 把当前子结点的值赋值给父结点
            i = k; // 把当前子结点作为新的父结点，继续向下循环比较是否还有子结点
        } else {
            break; // 直到以 i 父结点的树无子结点未比较为止
        }
    }
    //当for 循环结束后，我们已经将以i 为父结点的树的最大值，放在了 最顶(局部)
    arr[i] = temp; // 将temp值放到调整后的被交换的位置(子结点)
}