题目链接:https://pintia.cn/problem-sets/994805342720868352/problems/994805447855292416
题目大意:有n个学生,k个课程,给出每个课程的学生姓名,最后输出每个学生的选课情况。
分析:一开始用map+vector写的,不知道哪的错误,最后一个点一直报运行错误。后来看网上说是用hash,就是把学生姓名的前三个大写字母看成26进制,最后数字是十进制,转换成对应的编号就可以了。
#include <bits/stdc++.h>
using namespace std;
const int N = 180000;
int n, k;
string name;
vector<int> v[N];
int get_num(string s) {
int sum = 0;
for(int i = 0; i < 3; i++)
sum = sum * 26 + (s[i] - 'A');
sum = sum * 10 + (s[3] - '0');
return sum;
}
int main() {
scanf("%d %d", &n, &k);
for(int i = 1; i <= k; i++) {
int c, num;
scanf("%d %d", &c, &num);
for(int j = 1; j <= num; j++) {
cin >> name;
int id = get_num(name);
v[id].push_back(c);
}
}
for(int i = 1; i <= n; i++) {
cin >> name;
int id = get_num(name);
cout << name << ' ' << v[id].size();
if(v[id].size() > 0) {
sort(v[id].begin(), v[id].end());
printf(" ");
} else {
printf("\n");
continue;
}
for(int j = 0; j < v[id].size(); j++)
printf("%d%s", v[id][j], j == v[id].size() - 1 ? "\n" : " ");
}
return 0;
}