1039. Course List for Student (25)

来源:互联网 发布:淘宝如何看店铺所在地 编辑:程序博客网 时间:2024/06/16 02:22

    直接用 string 存储名字会由于C++的 cin, cout IO效率问题导致最后一个case超时,于是改为char数组存储名字,然后Hash为整数,再创建映射。

#include <iostream>#include <cstdio>#include <set>#include <unordered_map>#include <cstring>using namespace std;int Hash(char *name){return (name[0]-'A'+10)*36*36*36 + (name[1]-'A'+10)*36*36 + (name[2]-'A'+10)*36 + (name[3]-'0');}int main(){int n, k;scanf("%d%d", &n, &k);unordered_map<int, set<int>> stus;char name[5];for(int i = 0; i < k; ++i){int index, num;scanf("%d%d", &index, &num);for(int j = 0; j < num; ++j){scanf("%s", name);int val = Hash(name);if(stus.find(val) == stus.end()) stus.insert({val, set<int>()});stus[val].insert(index);}}for(int i = 0; i < n; ++i){scanf("%s", name);int val = Hash(name);printf("%s %d", name, (int)stus[val].size());for(auto& course : stus[val]){printf(" %d", course);}printf("\n");}return 0;}


0 0
原创粉丝点击