7-2 寻找大富翁 (25 分)
胡润研究院的调查显示,截至2017年底,中国个人资产超过1亿元的高净值人群达15万人。假设给出N个人的个人资产值,请快速找出资产排前M位的大富翁。
输入格式:
输入首先给出两个正整数N(≤106)和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);
}