堆排序讲解

给定 n 个整数 a1, a2, ..., an，使用堆排序的方法把这些数字按从小到大的顺序输出。

代码：

#include <bits/stdc++.h>
using namespace std;
const int N = 1e5 + 10;
int n, a[N];

//大顶堆
//arr为存储堆的数组
//len为数组的长度
//i为待维护节点的下标
void heapify(int arr[], int len, int i) {
    //令在当前堆，下标为i的值最大
    int largest = i;
    int lson = i * 2 + 1;
    int rson = (i + 1) * 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], arr[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], arr[0]);
        heapify(arr, i, 0);
    }
}

int main() {
    scanf("%d", &n);
    for(int i = 0; i < n; ++i)
        scanf("%d", &a[i]);
    heapsort(a, n);
    for(int i = 0; i < n; ++i)
        printf("%d ", a[i]);
    return 0;
}

样例：

输入：

5

2 3 5 4 1

输出：

1 2 3 4 5