PTA数据结构与算法题目集7-47 打印选课学生名单

7-47 打印选课学生名单

分数 25

作者 DS课程组

单位 浙江大学

假设全校有最多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

代码长度限制

16 KB

时间限制

1000 ms

内存限制

64 MB

栈限制

8192 KB

注意

本题使用cin和cout进行输入输出时，最后一个测试点会超时！

AC代码

#include "bits/stdc++.h"
using namespace std;

struct student {
    char name[5];
    int c;
}stu[40005];

struct lesson {
    int len;
    vector<string> per;
}les[2505];

int n,k;
int main() {
//    cin >> n >> k;
    scanf("%d%d",&n,&k);
    for(int i = 1; i <= n; i++) {
//        cin >> stu[i].name >> stu[i].c;
        scanf("%s%d",stu[i].name,&stu[i].c);
        for(int j = 1,x; j <= stu[i].c; j++) {
//            cin >> x;
            scanf("%d",&x);
            les[x].len++;
            les[x].per.push_back(stu[i].name);
        }
    }

    for(int i = 1; i <= k; i++) {
        printf("%d %d\n",i,les[i].len);
//        cout << i << " " << les[i].len << endl;
        sort(les[i].per.begin(), les[i].per.end());

        for(int j = 0; j < les[i].len; j++) {
//            c_str()函数使得变量类型变成char指针,即转化为C风格字符串
            printf("%s\n", les[i].per[j].c_str());
//            cout << les[i].per[j] << endl;

//            typeid().name() 返回变量的数据类型
//            cout << typeid(les[i].per[j].c_str()).name() << endl;
//            cout << typeid(les[i].per[j]).name() << endl;
        }
    }
    return 0;
}