PAT甲级 1047 Student List for Course (25分)

1047 Student List for Course (25分)

题目链接：PAT A 1047


题目大意：输入第一行给出两个数n和k，分别代表学生人数和课程总数，接下来n行，每行给出一个学生姓名（由三个大写字母和一个数字组成）和他所选的课程数以及课程编号（编号从1到k）。要求输出所有课程的选课人数以及学生姓名，并且学生姓名是按照字母序升序的规则进行输出的。

思路分析：与甲级1039题类似，这题数据量也比较大，因此使用string是很容易超时的（具体分析见1039题的题解），所以我们仍然使用字符数组。用一个vector数组来记录每门课程所选课的学生，对于每个输入的课程编号，向数组里push_back()第i个学生，这样输入结束后我们就得到了每个课程中选课的学生。接下来从第一门课程开始，对于每门课程中的学生进行排序，输出即可。

AC代码：

#include<iostream>
#include<cstring>
#include<vector>
#include<algorithm>
using namespace std;
char name[40010][5]; //存储学生的姓名 
bool cmp(int a, int b) { //排序 
	return strcmp(name[a], name[b]) < 0;
}
int main() {
	int n, k, num, courseid;
	scanf("%d %d", &n, &k);
	vector<int> course[2510]; //记录每门课程选课的学生 
	for(int i = 0; i < n; i++) {
		scanf("%s %d", name[i], &num);
		for(int j = 0; j < num; j++) {
			scanf("%d", &courseid);
			course[courseid].push_back(i);
		} 
	}
	for(int i = 1; i <= k; i++) {
		printf("%d %d\n", i, course[i].size());
		sort(course[i].begin(), course[i].end(), cmp); //排序 
		for(int j = 0; j < course[i].size(); j++)
			printf("%s\n", name[course[i][j]]);
	}
	return 0;
}