PAT1039
卡两样东西,一个是string的map,另一个是cin,cout
谢谢PAT让我感受到了OI的风采
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 }