【算法笔记】A1047 Student List for Course

https://pintia.cn/problem-sets/994805342720868352/problems/994805433955368960

题意

　　给出每个学生的选课情况，输出每节课选课的学生。

思路

　　参考了上机指南的代码，用char[n][5]存放学生姓名，vector<int> stulList[] 存放选每门课的学生编号，通过编号对应的姓名在vector数组里排序，妙啊

　　试了试用set能不能AC，发现不行，用set<string>会超时，用set<char*>又不能自动排序还要自定义排序函数，用set<int>要去掉自动排序用sort排序跟用vector<int>没啥区别。综上所述用vector是最省时省力的方法。

 code

#include<bits/stdc++.h>
using namespace std;
char name[40010][5];
vector<int> stuList[2510];
bool cmp(int a, int b){
    return strcmp(name[a], name[b]) < 0;
}
int main(){
    int n, k, c, courseId;
    scanf("%d%d", &n, &k);
    for(int i = 0; i < n; i++){
        scanf("%s %d", name[i], &c);
        for(int j = 0; j < c; j++){
            scanf("%d", &courseId);
            stuList[courseId].push_back(i);
        }
    }
    for(int i = 1; i <= k; i++){
        int size = stuList[i].size();
        printf("%d %d
", i, size);
        sort(stuList[i].begin(), stuList[i].end(), cmp);
        for(int j = 0; j < size; j++){
            printf("%s
", name[stuList[i][j]]);
        }
    }
    return 0;
}