【PAT-A】1055. The World's Richest 写题记录

我的看到题目的直接思路是先按年龄排序，挑选出年龄在区间内的，再按正常排序，有M个输出M个，没M个全部输出。这样因为排了2次序，有个大数据量的测试点会过不去。

sort排序的时候，注意&a，&a[0]，&a[0]+1，&a+1的区别

#include <cstdio>
#include <cstring>
#include <algorithm>
using namespace std;
struct People{
	char name[10];
	int age;
	int worth;
}peo[100020];

bool cmp1(People a,People b){
	if(a.worth!=b.worth) return a.worth>b.worth;
	else if(a.age!=b.age) return a.age<b.age;
	else return strcmp(a.name,b.name)<0;
}

bool cmp2(People a,People b){
	if(a.age!=b.age) return a.age<b.age;
	else return strcmp(a.name,b.name)<0;
}
int main(){
	int n,k;
	scanf("%d%d",&n,&k);
	for (int i=0;i<n;i++){
		scanf("%s",&peo[i].name);
		scanf("%d %d",&peo[i].age,&peo[i].worth);
	} 
	sort(peo,peo+n,cmp2);
	for (int i=1;i<=k;i++){
		int m,amin,amax;
		scanf("%d %d %d",&m,&amin,&amax);
		printf("Case #%d:\n",i);
		int tmin=-1,t=0;
		for (int j=0;j<n;j++){
			if (peo[j].age>=amin && peo[j].age<=amax){
				if(t==0) tmin = j;
				t++;
			}
		}
		if (t==0) printf("None\n");
		else {
			sort(&peo[tmin],&peo[tmin]+t,cmp1);
			if (t<m) m=t;
			for (int q=tmin;q-tmin<m;q++){
				printf("%s %d %d\n",peo[q].name,peo[q].age,peo[q].worth); 
			}
		}
	}
	return 0;
}

知道这点以后，马上就能分析出来，只一次排序，然后再按照年龄输出就好了。代码也更简单。

#include <cstdio>
#include <cstring>
#include <algorithm> 
using namespace std;
struct People{
	char name[10];
	int age;
	int worth;
}peo[100020];

bool cmp1(People a,People b){
	if(a.worth!=b.worth) return a.worth>b.worth;
	else if(a.age!=b.age) return a.age<b.age;
	else return strcmp(a.name,b.name)<0;
}

int main(){
	int n,k;
	scanf("%d%d",&n,&k);
	for (int i=0;i<n;i++){
		scanf("%s",&peo[i].name);
		scanf("%d %d",&peo[i].age,&peo[i].worth);
	} 
	sort(peo,peo+n,cmp1);
	for (int i=1;i<=k;i++){
		int m,amin,amax;
		scanf("%d %d %d",&m,&amin,&amax);
		printf("Case #%d:\n",i);
		int tmin=-1,t=0;
		for (int j=0;j<n;j++){
			if (t<m && peo[j].age>=amin && peo[j].age<=amax){
				printf("%s %d %d\n",peo[j].name,peo[j].age,peo[j].worth); 
				t++;
			}
		}
		if (t==0) printf("None\n");
	}
	return 0;
}