pat 1055 区间前k个

http://pat.zju.edu.cn/contests/pat-a-practise/1055

第二组数据比较大，如果单纯排序直接检索会超时，因为每次都是对所有数据进行遍历。

N/200=500,说明同一年龄最多可以有500个人，而M=100比较小，意味着同一年龄100以后的人都不会被搜到。

 

#include<iostream>
#include<cstring>
#include<cstdio>
#include <algorithm>
using namespace std;
struct node{
	char name[10];
	int age,worth;
}a[100005];
int b[20005];
int c[205];
bool cmp2(const node& p,const node& q){
	if(p.worth==q.worth)
	{
		if (p.age==q.age)
		{
			return strcmp(p.name,q.name)<=0;
		}
		return p.age<q.age;
	}
	return p.worth>q.worth;
}
int main()
{
	int n,q,i,M,ma,mb,txt=1,len;
	while(~scanf("%d %d",&n,&q))
	{
		memset(c,0,sizeof(c));
		for (i=0;i<n;++i){
			scanf("%s%d%d",a[i].name,&a[i].age,&a[i].worth);
		}
		sort(a,a+n,cmp2);
		for (len=i=0;i<n;++i){
			if (c[a[i].age]<=100)
			{
				b[len++]=i;
				c[a[i].age]++;
			}
		}
// 		printf("----\n");
// 		for (i=0;i<n;++i){
// 			printf("%s %d %d\n",a[b[i]].name,a[b[i]].age,a[b[i]].worth);
// 		}
		while(q--){
			scanf("%d%d%d",&M,&ma,&mb);
			if(ma>mb){
				ma^=mb;
				mb^=ma;
				ma^=mb;
			}
			printf("Case #%d:\n",txt++);
			int cnt=0;
			for (i=0;i<len&&cnt<M;++i){
				if(a[b[i]].age>=ma&&a[b[i]].age<=mb){
					printf("%s %d %d\n",a[b[i]].name,a[b[i]].age,a[b[i]].worth);
					cnt++;
				}
			}
			if(cnt==0){
				printf("None\n");
			}
		}
	}
	return 0;
}