寻找大富翁
Total Submission(s): 498 Accepted Submission(s): 282
Problem Description
浙江桐乡乌镇共有n个人,请找出该镇上的前m个大富翁.
Input
输入包含多组测试用例.
每个用例首先包含2个整数n(0<n<=100000)和m(0<m<=10),其中: n为镇上的人数,m为需要找出的大富翁数, 接下来一行输入镇上n个人的财富值.
n和m同时为0时表示输入结束.
每个用例首先包含2个整数n(0<n<=100000)和m(0<m<=10),其中: n为镇上的人数,m为需要找出的大富翁数, 接下来一行输入镇上n个人的财富值.
n和m同时为0时表示输入结束.
Output
请输出乌镇前m个大富翁的财产数,财产多的排前面,如果大富翁不足m个,则全部输出,每组输出占一行.
Sample Input
3 12 5 -15 31 2 3 4 50 0
Sample Output
55 4 3
Source
分析:
题目看起来就是排序问题,不过最初用插入排序(平均时间复杂度为O(n^2))会超时,选择
计数排序
: 使用一个额外数组a,a[i]表示A中值等于i的个数。因为存在财富值小于0的情况,所以a[i+5000]表示财富值为0的富翁的人数。然后从数组最大位9999开始输出不为0的元素坐标。平均时间复杂度为O(n+k)。
代码如下:
#include <iostream>
using namespace std;
int main()
{
int n,m;
int i,j,t,c;
int a[10000]={0};
while(1)
{
cin>>n>>m;
if(n==0 && m==0)break;
memset(a,0,sizeof(int)*10000);
for(i=0;i<n;i++)
{
cin>>t;
a[t+5000]++;
}
if(m>n)m=n;
c=0;
for(i=10000-1;i>=0 && c<m;i--)
{
if(a[i])
{
for(j=0;j<a[i] && c<m;j++)
{
cout<<i-5000;
if(c!=m-1)cout<<" ";
c++;
}
}
}
cout<<endl;
}
return 0;
}