PAT甲级 1055 The World's Richest (25分)

1055 The World’s Richest (25分)

题目链接：PAT A 1055


题目大意：输入第一行给出两个数n和k，分别代表人的总数和查询总数。随后n行给出一些人的名字，年龄，财富值。最后给出k次查询，每次查询给出一个数m以及年龄最小值和最大值，其中m是最大输出数（最大值100，即满足要求的可能有成千上万人，但是我们只输出100人）。要求我们输出在这个年龄区间内的m人，输出规则是按财富值降序输出，如果财富值相同，则按年龄升序输出，如果年龄也相同，则按字母序升序输出。题目保证不存在什么条件都一样的人。

思路分析：由于最大输出数是100，因此我们可以在排序后遍历一遍数组，将每个年龄的前99人存储到新的数组中，后面的查询直接在这个新数组中进行，这样可以显著降低数据量，不会导致超时。

AC代码：

#include<iostream>
#include<cstring>
#include<algorithm>
#include<vector>
using namespace std;
struct people{
	char name[10];
	int age, worth;
};
bool cmp(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, m, min, max, age[210] = {0};
	scanf("%d %d", &n, &k);
	vector<people> v(n), ans;
	for(int i = 0; i < n; i++) 
		scanf("%s %d %d", &v[i].name, &v[i].age, &v[i].worth);
	sort(v.begin(), v.end(), cmp);
	for(int i = 0; i < n; i++) { //统计每个年龄前100人 
		if(age[v[i].age] < 100) {
			ans.push_back(v[i]);
			age[v[i].age]++;
		}
	}
	for(int i = 1; i <= k; i++) {
		scanf("%d %d %d", &m, &min, &max);
		printf("Case #%d:\n", i);
		int count = 0; //统计输出人数 
		for(int j = 0; j < ans.size() && count < m; j++) {
			if(ans[j].age >= min && ans[j].age <= max) {
				printf("%s %d %d\n", ans[j].name, ans[j].age, ans[j].worth);
				count++;
			}
		}
		if(count == 0) //没有在这个年龄区间的人 
			printf("None\n");
	}
	return 0;
}