题目链接:https://www.acwing.com/problem/content/description/840/

时/空限制：1s / 64MB

题目描述

输入一个长度为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


解题思路

题意:从小到大输出前m小的数。

思路:这种题基本上都是先排个序，然后再把前m个输出。可是我们只需要求出前m个就行了，剩余的那些就不需要排了，所以我们很容易想到堆排序。首先建立一个小根堆，每次把最小的放到最后，求出前m小就结束。

Accepted Code:

/* 
 * @Author: lzyws739307453 
 * @Language: C++ 
 */
#include <bits/stdc++.h>
using namespace std;
const int MAXN = 1e5 + 5;
int spt[MAXN];
void PercDown(int *spt, int u, int size) {
    int t = u;
    if ((u << 1) <= size && spt[u << 1] < spt[t])
        t = u << 1;
    if ((u << 1 | 1) <= size && spt[u << 1 | 1] < spt[t])
        t = u << 1 | 1;
    if (t != u) {
        swap(spt[u], spt[t]);
        PercDown(spt, t, size);
    }
}
void Heap_Sort(int *spt, int n, int m) {
    for (int i = n / 2; i; i--)
        PercDown(spt, i, n);
    for (int i = n; i > n - m; i--) {
        swap(spt[1], spt[i]);
        PercDown(spt, 1, i - 1);
    }
}
int main() {
    int n, m;
    scanf("%d%d", &n, &m);
    for (int i = 1; i <= n; i++)
        scanf("%d", spt + i);
    Heap_Sort(spt, n, m);
    for (int i = n; i > n - m; i--)
        printf("%d ", spt[i]);
    printf("\n");
    return 0;
}