【题解-Acwing】838. 堆排序

题目：838. 堆排序

题目描述

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

输入

第一行包含整数 n 和 m。

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

输出

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

数据范围

1 ≤ m ≤ n ≤ 10⁵，
1 ≤ 数列中元素 ≤ 10⁹

时空限制

1s / 64MB

输入样例

5 3
4 5 1 3 2

输出样例

1 2 3

代码

#include<iostream>

using namespace std;

const int Maxn = 1e5+10;

int n, m;
int h[Maxn], sizee;

void down(int u){
    int t = u;
    if(u*2 <= sizee && h[u*2] < h[t]){
        t = u*2;
    }
    if(u*2+1 <= sizee && h[u*2+1] < h[t]){
        t = u*2+1;
    }
    if(u != t){
        swap(h[u], h[t]);
        down(t);
    }
}

int main(){
    scanf("%d%d", &n, &m);
    for(int i=1;i<=n;i++){
        scanf("%d", &h[i]);
    }
    sizee = n;
    for(int i=n/2;i;i--){
        down(i);
    }
    while(m--){
        printf("%d ", h[1]);
        h[1] = h[sizee];
        sizee --;
        down(1);
    }
    return 0;
}

结果