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

1039 Course List for Student (25分)

题目链接：PAT A 1039


题目大意：输入第一行给出两个数n和k，分别代表需要查询的学生人数和一共有多少门课（课程编号从1到k）。之后对于每一门课，给出选课人数和选课的学生姓名。最后一行给出n个学生的姓名，要求根据姓名查找这个学生选了哪些课，输出选课门数和课程编号。

思路分析：一开始以为是非常简单的一道STL容器题，即应用map<string, vector<int>>来解决问题。但是提交上去后有一个测试点运行超时，这是因为题中给的数据量比较庞大，可能有4万学生和2500门课程，这就导致用string输入有可能超时，所以不能应用map和string来解题，只能使用哈希映射来解题。由于题目中学生姓名由三个大写字母加一个数字构成，而字母A-Z可以用数字0-25来表示，这样就把26个大写字母对应到了26进制当中，之后按照其他进制转10进制的思路将其转换为10进制，所得到的id应该是唯一的。定义一个vector<int>stu[maxn]数组，中括号的意思是对于每一个stu[0],stu[1]等，本身又是一个变长数组。之后按照转换的id把对应的课程添加进去，查找的时候输出就可以了。

注意点1：maxn的值为26 * 26 * 26 * 10，因为前三个字母是26进制，最后一个数字是10进制。

注意点2：因为maxn的值过大，所以vector应该定义在全局变量中，定义在主函数中会运行错误。

注意点3：对于数据较大的题目，一般选择string很容易超时，应该使用字符数组进行字符串的存储。

AC代码：

#include<iostream>
#include<algorithm>
#include<vector>
using namespace std;
const int maxn = 26 * 26 * 26 * 10; 
int change(char name[]) { //将姓名转换为10进制 
	int id = 0;
	for(int i = 0; i < 3; i++) 
		id = id * 26 + (name[i] - 'A');
	id = id * 10 + (name[3] - '0');
	return id;
}
vector<int> stu[maxn]; //存储学生所选的课程 
int main() {
	int n, k, courseid, peoplenum;
	scanf("%d %d", &n, &k);
	char name[5];
	for(int i = 0; i < k; i++) {
		scanf("%d %d", &courseid, &peoplenum);
		for(int j = 0; j < peoplenum; j++) {
			scanf("%s", name);
			int id = change(name); //转换 
			stu[id].push_back(courseid);
		}
	}
	for(int i = 0; i < n; i++) {
		scanf("%s", name);
		int id = change(name);
		sort(stu[id].begin(), stu[id].end());
		printf("%s %d", name, stu[id].size());
		for(int j = 0; j < stu[id].size(); j++)
			printf(" %d", stu[id][j]);
		printf("\n");
	}
	return 0;
}