PAT-A 1039 Course List for Student (25 分)

浙江大学有 40000 名学生，提供 2500 门课程。

现在给定所有课程的报名学生姓名列表。

给定你一个学生名单，请你查询名单中的学生的具体选课情况。

输入格式
第一行包含两个整数 N 和 K，分别表示需要查询的学生数量以及课程数量。

接下来包含 K 个课程的具体报名信息。

每个课程信息的第一行包含两个整数 i 和 Ni，分别表示该课程的编号（所有课程编号从 1 到 K）以及报课人数。

第二行包含 Ni 个学生姓名，学生姓名由三个大写字母和一个数字组成。

最后一行包含 N 个学生姓名，表示要查询的学生名单。

输出格式
共 N 行，每行输出一个学生的姓名，报课总数，以及所有报名的课程的编号（按照升序排列）。

学生信息的输出顺序应与查询名单中的顺序一致。

数据范围
1≤N≤40000,
1≤K≤2500
1≤Ni≤200
输入样例：
11 5
4 7
BOB5 DON2 FRA8 JAY9 KAT3 LOR6 ZOE1
1 4
ANN0 BOB5 JAY9 LOR6
2 7
ANN0 BOB5 FRA8 JAY9 JOE4 KAT3 LOR6
3 1
BOB5
5 9
AMY7 ANN0 BOB5 DON2 FRA8 JAY9 KAT3 LOR6 ZOE1
ZOE1 ANN0 BOB5 JOE4 JAY9 FRA8 DON2 AMY7 KAT3 LOR6 NON9
输出样例：
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
NON9 0

哈希表加一个set模拟一下

#include <bits/stdc++.h>

using namespace std;
unordered_map<string,set<int> > mp;
int n,m;
int main()
{
    ios::sync_with_stdio(false);
    cin >> n >>m;
    while(m--)
    {
        int k,course_id;
        string stu;
        cin >> course_id;
        cin >> k;
        while(k--)
        {
            cin >> stu;
            mp[stu].insert(course_id);
        }
    }
    while(n--)
    {
        string stu;
        cin >> stu;
        cout <<stu<<" "<<mp[stu].size();
        for(int each:mp[stu])
        {
            cout <<" "<<each;
        }
        if(n!=0)
        cout <<endl;
    }
    return 0;
}