PAT1039

 1 #include <iostream>
 2 #include <vector>
 3 #include <string>
 4 #include <fstream>
 5 #include <map>
 6 #include <algorithm>
 7 
 8 using namespace std;
 9 
10 //#define OJ
11 
12 #ifdef OJ
13 #define fin cin
14 #endif
15 
16 int hash_str(const char *str){
17     return (str[0] - ‘A‘) * 26 * 26 * 10 + (str[1] - ‘A‘) * 26 * 10 + (str[2] - ‘A‘) * 10 + (str[3] - ‘0‘);
18 }
19 
20 vector<int> student_course[26 * 26 * 26 * 10];
21 
22 int main(){
23 #ifndef OJ
24     ifstream fin("in.data");
25 #endif
26 
27     ios::sync_with_stdio(NULL);
28 
29     int N, K;
30     fin >> N >> K;
31 
32     char name[5];
33 
34     for (int i = 0; i < K; i++){
35         int idx, num;
36         fin >> idx >> num;
37 
38         for (int j = 0; j < num; j++){
39             fin >> name;
40             student_course[hash_str(name)].push_back(idx);
41         }
42     }
43 
44     for (int i = 0; i < N; i++){
45         fin >> name;
46 
47         vector<int> &course_idx = student_course[hash_str(name)];
48         sort(course_idx.begin(), course_idx.end());
49 
50         printf("%s %d", name, course_idx.size());
51         for (int j = 0; j < course_idx.size(); j++){
52             printf(" %d", course_idx[j]);
53         }
54 
55         printf("\n");
56     }
57 
58     return 0;
59 }