PAT甲级真题1039 学生课程列表(STL的运用）

浙江大学有 4000040000 名学生，提供 25002500 门课程。

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

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

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

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

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

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

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

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

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

数据范围
1≤N≤400001≤N≤40000,
1≤K≤25001≤K≤2500
1≤Ni≤2001≤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

#include <iostream>
#include <algorithm>
#include <unordered_map>
#include <vector>

using namespace std;

unordered_map<string, vector<int>> students;

int main()
{
    ios::sync_with_stdio(false);
    cin.tie(0);

    int n, k;
    cin >> n >> k;

    while (k -- )
    {
        int id, m;
        cin >> id >> m;
        while (m -- )
        {
            string name;
            cin >> name;
            students[name].push_back(id);
        }
    }

    while (n -- )
    {
        string name;
        cin >> name;
        vector<int>s=students[name];
        cout << name << ' ' << s.size();
        sort(s.begin(), s.end());

        for (vector<int>::iterator it=s.begin();it!=s.end();it++) cout << ' ' << *it;
        cout << endl;
    }

    return 0;
}