PAT (Advanced Level) Practice A1047 Student List for Course （25 分）(C++)（甲级）（vector）

原题链接1047 Student List for Course

#include<algorithm>
#include<iostream>
#include<cstdio>
#include<cstring>
#include <ctime>
#include <cmath>
#include <vector>
using namespace std;

const int MAX_N = 40010;//学生总数
const int MAX_M = 2510;//课程总数
char Name[MAX_N][5];//存放每个学生的名字
vector<int> Course[MAX_M];//存放每个学生所选择的课程号

bool cmp(int A, int B)//排序
{
    return strcmp(Name[A], Name[B]) < 0;
}

int main()
{
    int N, K;
    scanf("%d %d", &N, &K);
    for(int i=0, j; i<N; i++)
    {
        char stu_name[5];
        int C, course_id;
        scanf("%s %d", Name[i], &C);
        for(int j=0; j<C; j++)//输入
        {
            scanf("%d", &course_id);
            Course[course_id].push_back(i);
        }
    }
    for(int i=1; i<=K; i++)
    {
        printf("%d %d\n", i, Course[i].size());//输出课程号和选择该课程的学生总数
        sort(Course[i].begin(), Course[i].end(), cmp);//排序
        for(int j=0; j<Course[i].size(); j++)
        {
            printf("%s\n", Name[Course[i][j]]);//输出名字
        }
    }
    return 0;
}