AcWing - 堆排序(堆排序求前m小的数)

题目链接: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;
}
展开阅读全文
©️2020 CSDN 皮肤主题: 代码科技 设计师: Amelia_0503 返回首页
实付0元
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、C币套餐、付费专栏及课程。

余额充值