PAT1039

卡两样东西，一个是string的map，另一个是cin，cout

谢谢PAT让我感受到了OI的风采

#include <iostream>
#include <vector>
#include <string>
#include <fstream>
#include <map>
#include <algorithm>

using namespace std;

//#define OJ

#ifdef OJ
#define fin cin
#endif

int hash_str(const char *str){
    return (str[0] - 'A') * 26 * 26 * 10 + (str[1] - 'A') * 26 * 10 + (str[2] - 'A') * 10 + (str[3] - '0');
}

vector<int> student_course[26 * 26 * 26 * 10];

int main(){
#ifndef OJ
    ifstream fin("in.data");
#endif

    ios::sync_with_stdio(NULL);

    int N, K;
    fin >> N >> K;

    char name[5];

    for (int i = 0; i < K; i++){
        int idx, num;
        fin >> idx >> num;

        for (int j = 0; j < num; j++){
            fin >> name;
            student_course[hash_str(name)].push_back(idx);
        }
    }

    for (int i = 0; i < N; i++){
        fin >> name;

        vector<int> &course_idx = student_course[hash_str(name)];
        sort(course_idx.begin(), course_idx.end());

        printf("%s %d", name, course_idx.size());
        for (int j = 0; j < course_idx.size(); j++){
            printf(" %d", course_idx[j]);
        }

        printf("
");
    }

    return 0;
}