堆排序--数组模拟

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

输入格式

第一行包含整数 nn 和 mm。

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

输出格式

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

数据范围

1≤m≤n≤1051≤m≤n≤105，
1≤数列中元素≤1091≤数列中元素≤109

输入样例：

5 3
4 5 1 3 2

输出样例：

1 2 3

 

#include<iostream>
using namespace std;
const int N=1e5+10;
int h[N];
int n;//建立小根堆 

//up建立堆模板 

/* 
void up(int u)
{
	while(u/2&&h[u/2]>h[u])
	{
		swap(h[u/2],h[u]);
		u/=2;
	}
	return ;
}
*/

//down建立堆模板 
void down(int u)
{
	int t=u;
	if(u*2<=n&&h[u*2]<h[t]) t=u*2;
	if(u*2+1<=n&&h[u*2+1]<h[t]) t=u*2+1;
	if(u!=t)
	{
		swap(h[u],h[t]);
		down(t);
	}
	return ;
}

int main()
{
	cin>>n;int m;cin>>m;
	for(int i=1;i<=n;i++) cin>>h[i];
	
	for(int i=n/2;i;i--)
	{
		down(i);
	}
	
	while(m--)
	{
		cout<<h[1]<<" ";
		h[1]=h[n];
		n--;
		down(1);
	}
	
	return 0;
}