PTA 习题集5-18 打印选课学生名单(哈希)

假设全校有最多40000名学生和最多2500门课程。现给出每个学生的选课清单，要求输出每门课的选课学生名单。

输入格式:

输入的第一行是两个正整数：N（≤40000），为全校学生总数；K（≤2500），为总课程数。此后N行，每行包括一个学生姓名（3个大写英文字母+1位数字）、一个正整数C（≤20）代表该生所选的课程门数、随后是C个课程编号。简单起见，课程从1到K编号。

输出格式:

顺序输出课程1到K的选课学生名单。格式为：对每一门课，首先在一行中输出课程编号和选课学生总数（之间用空格分隔），之后在第二行按字典序输出学生名单，每个学生名字占一行。

输入样例:

10 5
ZOE1 2 4 5
ANN0 3 5 2 1
BOB5 5 3 4 2 1 5
JOE4 1 2
JAY9 4 1 2 5 4
FRA8 3 4 2 5
DON2 2 4 5
AMY7 1 5
KAT3 3 5 4 2
LOR6 4 2 4 1 5

输出样例:

1 4
ANN0
BOB5
JAY9
LOR6
2 7
ANN0
BOB5
FRA8
JAY9
JOE4
KAT3
LOR6
3 1
BOB5
4 7
BOB5
DON2
FRA8
JAY9
KAT3
LOR6
ZOE1
5 9
AMY7
ANN0
BOB5
DON2
FRA8
JAY9
KAT3
LOR6
ZOE1

题意

如上

题解

哈希AAA0--ZZZ9 总个数26*26*26*10，暴力的话用sort比较会超时

代码

 #include<stdio.h>
 #include<algorithm>
 #include<vector>
 using namespace std;

 vector<int> vec[];
 int main(){
     int n,k,c,cla,len[]={};
     char a[];
     scanf("%d%d",&n,&k);
     for(int i=;i<=n;++i){
         scanf("%s%d",a,&c);
         int N=(a[]-'A')*+(a[]-'A')*+(a[]-'A')*+(a[]-'');
         for(int j=;j<=c;++j){
             scanf("%d",&cla);
             vec[cla].push_back(N);
             ++len[cla];
         }
     }
     for(int i=;i<=k;++i){
         sort(vec[i].begin(),vec[i].end());
         printf("%d %d\n",i,len[i]);
         for(vector<int>::iterator it=vec[i].begin();it!=vec[i].end();++it){
             int a1,a2,a3,a4;
             a1=*it/;
             a2=(*it-a1*)/;
             a3=(*it-a1*-a2*)/;
             a4=*it%;
             printf("%c%c%c%C\n",a1+'A',a2+'A',a3+'A',a4+'');
         }
     }
     return ;
 }