838. 堆排序

 题目来源

https://www.acwing.com/problem/content/840/

//如何手写一个堆？完全二叉树 5个操作
//1. 插入一个数         heap[ ++ size] = x; up(size);
//2. 求集合中的最小值   heap[1]
//3. 删除最小值         heap[1] = heap[size]; size -- ;down(1);
//4. 删除任意一个元素   heap[k] = heap[size]; size -- ;up(k); down(k);
//5. 修改任意一个元素   heap[k] = x; up(k); down(k);
#include <iostream>
using namespace std;

int const N = 100010;
//h[i] 表示第i个结点存储的值，i从1开始，2*i是左子节点，2*i + 1是右子节点
//size 既表示堆里存储的元素个数，又表示最后一个结点的下标
int h[N], siz; //堆有两个变量h[N]，size; 怎么这里的size和文件里有冲突，只能改成siz了

void down(int u)
{
    int t = u;//t存储三个结点中存在的最小的结点的下标，初始化为当前结点u
    if (u * 2 <= siz && h[u * 2] < h[t]) t = u * 2; // 左子节点存在并且小于当前结点，更新t的下标
    if (u * 2 + 1 <= siz && h[u * 2 + 1] < h[t]) t = u * 2 + 1;//右子节点存在并且小于当前结点，更新t的下标
    if (t != u)//如果t==u意味着不用变动，u就是三个结点中拥有最小值的结点下标，否则交换数值
    {
        swap(h[t], h[u]);
        down(t); //交换数值后，t这个结点存储原本u的值，u存储存储t的值（三个数中的最小值）。u不用调整了，但t情况不明，可能需要调整。直到它比左右子节点都小
    }
}

int main()
{
    int n, m;
    cin >>n>>m;
    for (int i = 1; i <= n; i ++ ) scanf("%d", &h[i]); 
    siz = n; //初始化size,表示堆里有n 个元素
    for (int i = n / 2; i; i --) down(i); //把堆初始化成小根堆，从二叉树的倒数第二行开始，把数字大的下沉
    while (m -- )
    {
        printf("%d ", h[1]);
        h[1] = h[siz];
        siz --;
        down(1);
    }

    return 0;
}

#include<iostream>
using namespace std;

const int N = 1e5+10;

int n , m;
int h[N],cnt;

// -----------------------------模板开始--------------------
// 这个模板的作用是父节点和其两个孩子节点的最小值，再交换父节点和这个最小值
void down(int u){
    int t = u;
    if(u*2 <= cnt && h[u*2] < h[t]) t = u*2;
    if(u*2 + 1 <= cnt && h[u*2+1] < h[t]) t = u*2+1;
    if(u!=t){
        swap(h[u],h[t]);
        down(t); // 递归处理这个输入的值，直到它down到不能down为止，即整棵树又变成了一个小根堆
    }
}
// -----------------------------模板结束--------------------

int main(){
    scanf("%d%d",&n,&m);
    for(int i = 1;i <= n;i++) scanf("%d",&h[i]); // h[]来存输入的n个节点，注意，节点的下标要从1开始

    cnt = n; 
    /*
        cnt来存节点的最大下标,因为找左右孩子需要用2*u和2*u+1，
        为了防止边界值问题，需要判断2*u < n,2*u+1<n;
        并且，cnt也可以看成指向堆尾的一个指针。
    */ 

    // 用down函数将整棵树调整为一个小根堆
    for(int i = n / 2;i;i--) down(i); // 从2分之n开始down，建堆，时间复杂度为O(n)


    while(m--){
        printf("%d ",h[1]); // 输出堆顶的数，即为当前数列的最小值
        h[1] = h[cnt--];  // 将堆最后一个数放到堆顶，cnt--使得在逻辑上删除这个数
        down(1); // 再将堆顶down下去，使得整棵树又变成了一个小根堆
    }
    return 0;
}