SDUT OJ 数据结构实验之排序四：寻找大富翁

数据结构实验之排序四：寻找大富翁

Time Limit: 200 ms Memory Limit: 512 KiB

Submit Statistic Discuss

Problem Description

2015胡润全球财富榜调查显示，个人资产在1000万以上的高净值人群达到200万人，假设给出N个人的个人资产值，请你快速找出排前M位的大富翁。

Input

首先输入两个正整数N( N ≤ 10^6)和M(M ≤ 10)，其中N为总人数，M为需要找出的大富翁数目，接下来给出N个人的个人资产，以万元为单位，个人资产数字为正整数，数字间以空格分隔。

Output

一行数据，按降序输出资产排前M位的大富翁的个人资产值，数字间以空格分隔，行末不得有多余空格。

Sample Input

6 3
12 6 56 23 188 60

Sample Output

188 60 56

Hint

请用堆排序完成。

#include <iostream>
#include<bits/stdc++.h>
using namespace std;
int m,n;
int a[234823];
void change(int i)
{
    int flag = 0;
    while(i*2<=m&&flag==0)
    {
        int t;
        if(a[i]<a[i*2])
            t = i;
        else t = i*2;
        if(i*2+1<=m&&a[i*2+1]<a[t])
            t = i*2+1;
        if(t!=i)
        {
            swap(a[i],a[t]);
            i = t;
        }
        else flag =1;
    }
}
void dui_sort()
{
    while(m>1)
    {
        swap(a[1],a[m]);
        m--;
     //   change(m);
    }
}
int main()
{
     cin>>n>>m;
        int nn =m;
        for(int i=1; i<=m; i++)
        {
            scanf("%d",&a[i]);
        }
        for(int i=m/2; i>=1; i--)
        {
            change(i);
        }
        int x;
        for(int i=m+1; i<=n; i++)
        {
            scanf("%d",&x);
           // cout<<a[1]<<endl;
            if(x>a[1])
            {
                a[1] = x;
               change(1);
//                for(int j=m/2;j>=1;j--)
//                {
//                    change(j);
//                }
            }
        }
//        for(int i=1; i<=nn; i++)
//    {
//        printf("%d",a[i]);
//        if(i!=nn)printf(" ");
//        else printf("\n");
//    }
      dui_sort();

    for(int i=1; i<=nn; i++)
    {
        printf("%d",a[i]);
        if(i!=nn)printf(" ");
        else printf("\n");
    }
    return 0;
}