7-2 寻找大富翁 (25 分)

7-2 寻找大富翁 (25 分)

胡润研究院的调查显示,截至2017年底,中国个人资产超过1亿元的高净值人群达15万人。假设给出N个人的个人资产值,请快速找出资产排前M位的大富翁。

输入格式:

输入首先给出两个正整数N(≤10​6​​)和M(≤10),其中N为总人数,M为需要找出的大富翁数;接下来一行给出N个人的个人资产值,以百万元为单位,为不超过长整型范围的整数。数字间以空格分隔。

输出格式:

在一行内按非递增顺序输出资产排前M位的大富翁的个人资产值。数字间以空格分隔,但结尾不得有多余空格。

输入样例:

8 3
8 12 7 3 20 9 5 18

输出样例:

20 18 12

堆排序

#include <iostream>
#include <algorithm>
#include <cstdio>
#include <map>
#include <stack>
#include <set>
#include <vector>
#include <cstring>
#include <queue>
#include <cmath>
#define FRER() freopen("in.txt","r",stdin);
#define FREW() freopen("out.txt","w",stdout);
#define mem(a,b) memset(a,b,sizeof(a));
#define go int T;scanf("%d",&T);for(int cas=1;cas<=T;cas++)
#define mod 1000000007
using namespace std;
typedef pair<int,int> pii;
typedef long long ll;
const int maxn = 1000000 + 7;
int a[maxn];
void heapAdjust(int i,int n){
    int tmp = a[i];
    for(int j=i<<1;j<=n;j<<=1){
        if(j<n&&a[j]<a[j+1]) j++;
        if(a[j]<a[i]) break;
        swap(a[i],a[j]);
        i=j;
    }
}
void heapSort(int n,int k){
    for(int i=n/2;i>=1;i--) heapAdjust(i, n);
    for(int i=n;i>=1&&k--;i--){
        if(i!=n) cout<<" ";
        cout<<a[1];
        swap(a[1], a[i]);
        heapAdjust(1, i-1);
    }
    cout<<endl;
}
int main() {
    //FRER();
    int n,k;
    cin>>n>>k;
    for(int i=1;i<=n;i++){
       scanf("%d",&a[i]);
    }
    heapSort(n,k);

}

猜你喜欢

转载自blog.csdn.net/Insist_77/article/details/84853656