输入一个长度为 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是节点总数。