0
点赞
收藏
分享

微信扫一扫

堆排序(模板)

洲行 2022-03-30 阅读 56
排序算法

输入一个长度为 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
难度:简单
时/空限制:1s / 64MB
总通过数:31481
总尝试数:48424
来源:模板题,AcWing
算法标签

#include<iostream>
using namespace std;
const int N = 1e5+10;
int n,m;
int h[N],num;
void down(int x)
{
    int t=x;
    if(x*2<=num&&h[x*2]<h[t]) t=x*2;
    if(x*2+1<=num&&h[x*2+1]<h[t]) t=x*2+1;
    if(t!=x)
    {
        swap(h[t],h[x]);
        down(t);
    }
}
int main()
{
    scanf("%d%d", &n, &m);
    for (int i = 0; i < n; i ++ ) scanf("%d", &h[i]);
    num=n;
    for (int i = n/2;i>=0; i --) down(i);
    while(m--)
    {
        cout << h[1]<<" ";
        h[1]=h[num--];
        down(1);
    }
    
}
举报

相关推荐

0 条评论