题目链接:https://pintia.cn/problem-sets/1101307589335527424/problems/1101314114875633664
题意:输入n个学生,k門课程,每门课程的编号n1和人数n2,并输入这n2个人的姓名,姓名由3个大写字母和1个10进制数组成。要求来询问的n个学生所选的课程编号,并按编号从小到大输出。
思路:首先利用进制思想将每个人的名字转换成10进制数,每一位的权分别为26,26,26,10,则编号最大为26×26×26×10+26×26×10+26×10+10=182790。之后可以设一个vector<int> v[183000],存储每个学生所选的课程。之后遍历k门课程就行了,另外虽然题目说了按照课程编号从小到大遍历,但好像并不是这样,还是需要sort一下,不然第2个测试点会wa。
AC代码:
1 #include2 using namespace std; 3 4 int n,k; 5 vector v[183000]; 6 7 int main(){ 8 scanf("%d%d",&n,&k); 9 for(int i=1;i<=k;++i){10 int n1,n2,t;11 char tmp[5];12 scanf("%d%d",&n1,&n2);13 while(n2--){14 scanf("%s",tmp);15 t=(((tmp[0]-'A')*26+(tmp[1]-'A'))*26+(tmp[2]-'A'))*10+(tmp[3]-'0');16 v[t].push_back(n1);17 }18 }19 while(n--){20 int t;21 char tmp[5];22 scanf("%s",tmp);23 printf("%s",tmp);24 t=(((tmp[0]-'A')*26+(tmp[1]-'A'))*26+(tmp[2]-'A'))*10+(tmp[3]-'0');25 int sz=v[t].size(); 26 printf(" %d",sz);27 sort(v[t].begin(),v[t].end());28 for(int i=0;i