堆排序（小根堆模板）

输入一个长度为 n 的整数数列，从小到大输出前 m 小的数。

输入格式

第一行包含整数 n 和 m。

第二行包含 n 个整数，表示整数数列。

输出格式

共一行，包含 m 个整数，表示整数数列中前 m 小的数。

数据范围

1≤m≤n≤10^5，
1≤数列中元素≤10^9

输入样例：

5 3
4 5 1 3 2

输出样例：

1 2 3

堆排序分为大根堆和小根堆，这里实现的是小根堆，定义是父节点比它的左右孩子的值都要小。

堆排序主要实现两个操作：down和up，把最小值调整到根节点，把最大值调整到最下面

堆排序的五种操作：

 

示例代码：

#include<iostream>
#include<algorithm>
using namespace std;

const int N=1e5+10;

int n,m;
int h[N],Size;  //h表示堆，size表示堆的大小

void down(int u)  //小根堆核心，向下比较，使父节点小于左右孩子
{
    int t=u;  //让t作为最小值
    if(u*2 <= Size && h[u*2] < h[t]) t=u*2; //左儿子存在且左儿子小于父节点，左儿子就是t
    if(u*2+1 <= Size && h[u*2+1] < h[t]) t=u*2+1; //右儿子更小，右儿子就是t
    if(u != t)  //如果父节点不是最小的
    {
        swap(h[u],h[t]); //和最小的交换,使父节点最小
        down(t); //对父节点与子节点中最小的位置（也就是刚刚交换完之后，原来的父节点现在所在的位置）进行递归向下操作，直到这个节点满足小根堆性质
    }
}
int main()
{
    ios::sync_with_stdio(0),cin.tie(0),cout.tie(0);
    cin>>n>>m;
    for(int i=1;i<=n;i++) cin>>h[i];
    Size=n;
    
    for(int i=n/2;i;i--) down(i); //堆排序，只需要对倒数第二层及上面的元素排序即可（有子节点的点），最下面的没有子节点进行比较，就不用进行down操作
    
    while(m--)
    {
        cout<<h[1]<<" ";  //按顺序从小到大输出，最小的就是根节点
        h[1]=h[Size]; //用最后的元素覆盖第一个
        Size--; //删除最后一个点
        down(1); //对第一个元素进行堆排序
    }
    return 0;
}

 

 关于down函数：

void down(int u)  //小根堆核心，向下比较，使父节点小于左右孩子
{
    int t=u;  //让t作为最小值
    if(u*2 <= Size && h[u*2] < h[t]) t=u*2; //左儿子存在且左儿子小于父节点，左儿子就是t
    if(u*2+1 <= Size && h[u*2+1] < h[t]) t=u*2+1; //右儿子更小，右儿子就是t
    if(u != t)  //如果父节点不是最小的
    {
        swap(h[u],h[t]); //和最小的交换,使父节点最小
        down(t); //对父节点与子节点中最小的位置（也就是刚刚交换完之后，原来的父节点现在所在的位置）进行递归向下操作，直到这个节点满足小根堆性质
    }
}

如果没有发生交换，说明父子节点都满足堆的性质（父节点最小），如果交换了，则意味着下面的所有点也要跟着变化，所以是对发生了交换的父节点（现在父子里面最小的成了父节点，而原来的父节点到了子节点的位置，这里是指原来的父节点） 进行递归操作，直到这个交换的父节点和它下面的子节点都满足堆的性质。

关于for循环堆排序的解释：

for(int i=n/2;i;i--) down(i); //堆排序，只需要对倒数第二层及上面的元素排序即可（有子节点的点），最下面的没有子节点进行比较，就不用进行down操作

因为堆排是对父节点和它的两个子节点排序，所以没有子节点的点就不用堆排，堆是一个完全二叉树，有子节点的点下标从1到n/2，n是节点总数。