PTA 7-49 打印学生选课清单 (25 point(s))

假设全校有最多40000名学生和最多2500门课程。现给出每门课的选课学生名单，要求输出每个前来查询的学生的选课清单。

输入格式:

输入的第一行是两个正整数：N（≤40000），为前来查询课表的学生总数；K（≤2500），为总课程数。此后顺序给出课程1到K的选课学生名单。格式为：对每一门课，首先在一行中输出课程编号（简单起见，课程从1到K编号）和选课学生总数（之间用空格分隔），之后在第二行给出学生名单，相邻两个学生名字用1个空格分隔。学生姓名由3个大写英文字母+1位数字组成。选课信息之后，在一行内给出了N个前来查询课表的学生的名字，相邻两个学生名字用1个空格分隔。

输出格式:

对每位前来查询课表的学生，首先输出其名字，随后在同一行中输出一个正整数C，代表该生所选的课程门数，随后按递增顺序输出C个课程的编号。相邻数据用1个空格分隔，注意行末不能输出多余空格。

输入样例:

10 5
1 4
ANN0 BOB5 JAY9 LOR6
2 7
ANN0 BOB5 FRA8 JAY9 JOE4 KAT3 LOR6
3 1
BOB5
4 7
BOB5 DON2 FRA8 JAY9 KAT3 LOR6 ZOE1
5 9
AMY7 ANN0 BOB5 DON2 FRA8 JAY9 KAT3 LOR6 ZOE1
ZOE1 ANN0 BOB5 JOE4 JAY9 FRA8 DON2 AMY7 KAT3 LOR6

结尾无空行

输出样例:

ZOE1 2 4 5
ANN0 3 1 2 5
BOB5 5 1 2 3 4 5
JOE4 1 2
JAY9 4 1 2 4 5
FRA8 3 2 4 5
DON2 2 4 5
AMY7 1 5
KAT3 3 2 4 5
LOR6 4 1 2 4 5

结尾无空行

 

// 桶排序 数组
#include <bits/stdc++.h>
using namespace std;

int name_change(char name[5]){
	return (name[0] - 'A') * 26 * 26 * 10 +
            (name[1] - 'A') * 26 * 10 +
            (name[2] - 'A') * 10 + 
            name[3] - '0';
}
 
main(void){
    vector<int> stu[26 * 26 * 26 * 10 + 1];
	int n, k, course, num, nameIndex;
	char name[5];
	cin >> n >> k;
	
	for (int i = 0; i < k; i++){
        // 读取课程编号及选课人数
		cin >> course >> num;
        
		for (int j = 0; j < num; j++){
            // 读取选课学生
			cin >> name;
            
            // 将名字转换为数字下标 将课程推入向量下标处
			nameIndex = name_change(name);
			stu[nameIndex].push_back(course);
		}
	}
    
	for (int i = 0; i < n; i++){
        // 读取名字 转化为下标 输出学生名字以及课程数量
		cin >> name;
		nameIndex = name_change(name);
        cout << name << " " << stu[nameIndex].size();
        
        // 对学生课程编号排序
		sort(stu[nameIndex].begin(), stu[nameIndex].end());
        
        // 输出课程编号
		for (int j = 0; j < stu[nameIndex].size(); j++)
            cout << " " << stu[nameIndex][j];
        
        // 下一个学生 换行
		cout << endl;
	}
}

#include <bits/stdc++.h>
using namespace std;

main(void){
    map<string, vector<int>> stu;
    int n, k;
    char name[5];
    // cin cout 会超时 换回 scanf printf
    scanf("%d %d", &n, &k);
    
    while(k--){
        // 读取课程编号及选课人数
        int course, num;
        scanf("%d %d", &course, &num);
        while(num--){
            // 读取选课学生
            scanf("%s", name);
            stu[name].push_back(course);
        }
    }
    
    while(scanf("%s", name) != EOF){
        printf("%s %d", name, stu[name].size());
        // 对学生课程编号排序
        sort(stu[name].begin(), stu[name].end());
        // 输出课程编号
        for(int i = 0; i < stu[name].size(); i++){
            printf(" %d", stu[name][i]);
        }
        // 下一个学生 换行
        printf("\n");
    }
}

参考代码1        参考代码2

记录在PAT编译器上日常谜之错误，仅仅是这两处复制粘贴一下就可以让结果从超时变成AC。我唯一能够看出来的区别一个是scanf前面是四个点一个是箭头。