对应HDU题目:点击打开链接
sort
Time Limit: 6000/1000 MS (Java/Others) Memory Limit: 65536/32768 K (Java/Others)Total Submission(s): 28012 Accepted Submission(s): 8491
Problem Description
给你n个整数,请按从大到小的顺序输出其中前m大的数。
Input
每组测试数据有两行,第一行有两个数n,m(0<n,m<1000000),第二行包含n个各不相同,且都处于区间[-500000,500000]的整数。
Output
对每组测试数据按从大到小的顺序输出前m大的数。
Sample Input
5 3 3 -35 92 213 -644
Sample Output
213 92 3
听说是一定范围内最快的排序方法。。。
#include<cstdio>
#include<cstdlib>
#include<cmath>
#include<map>
#include<queue>
#include<stack>
#include<vector>
#include<algorithm>
#include<cstring>
#include<string>
#include<iostream>
const int MAXN=1000000+10;
using namespace std;
int a[MAXN];
int c[MAXN];
int rank[MAXN];
int main()
{
//freopen("in.txt","r",stdin);
int n,k;
while(scanf("%d%d", &n,&k)==2)
{
memset(c,0,sizeof(c));
memset(rank,0,sizeof(rank));
for(int i=0; i<n; i++){
scanf("%d", &a[i]);
++c[a[i]+500001];//右移
}
for(int i=1; i<MAXN; i++) c[i]+=c[i-1];
for(int i=n-1; i>=0; i--) rank[--c[a[i]+500001]]=a[i];
printf("%d", rank[n-1]);
for(int i=n-2; i>=n-k; i--)
printf(" %d", rank[i]);
printf("\n");
}
return 0;
}
空间优化:
#include<cstdio>
#include<cstdlib>
#include<cmath>
#include<map>
#include<queue>
#include<stack>
#include<vector>
#include<algorithm>
#include<cstring>
#include<string>
#include<iostream>
const int MAXN=1000000+10;
using namespace std;
int a[MAXN];
int c[MAXN];
int main()
{
//freopen("in.txt","r",stdin);
int n,k;
while(scanf("%d%d", &n,&k)==2)
{
memset(c,0,sizeof(c));
for(int i=0; i<n; i++){
scanf("%d", &a[i]);
++c[a[i]+500000];//右移
}
int j=0;
for(int i=0; i<MAXN; i++)
while((c[i]--) > 0)
{
a[j++]=i-500000;
}
printf("%d", a[n-1]);
for(int i=n-2; i>=n-k; i--)
printf(" %d", a[i]);
printf("\n");
}
return 0;
}