PAT-A 1047. Student List for Course (25)

题目链接在此。

同样是一道STL(vector)的练习题。
先看看这个题：PAT-A 1039. Course List for Student (25)
看题目就应该知道这两题是镜像题目。

由上一题的结论可知，这一题应该以课程为主体。
在上一题的基础上，这题就变得十分简单了，直接看代码把。

#include<stdio.h>
#include<cstring>
#include<vector>
#include<algorithm>

using namespace std;

const int M = 2505;     //总课程数
const int N = 40005;    //总学生数 
vector<int> stuList[M]; //某课程的学生列表 
char name[N][5];       //学生姓名 

int getStuId(char name[]){
    int id = 0;
    for(int i = 0; i < 3; i++){
        id = id*26 + (name[i]-'A');
    }
    id = id*10 + name[3];

    return id;
}

bool cmp(int a, int b){
    return strcmp(name[a], name[b]) < 0;
}

int main(){

    int n,k;     //n 学生总数， k 课程总数 
    scanf("%d %d",&n, &k);

    int courseNum, courseId; //选课数目，课程编号 
    for(int i = 0; i < n; i++){
        scanf("%s",name[i]);
        scanf("%d", &courseNum);
        for(int j = 0; j < courseNum; j++){
            scanf("%d", &courseId);
            stuList[courseId].push_back(i);
        }
    } 

    for(int i = 1; i < k+1; i++){
        printf("%d %d\n", i, stuList[i].size());
        sort(stuList[i].begin(), stuList[i].end(), cmp);
        for(int j = 0; j < stuList[i].size(); j++){
            printf("%s\n",name[stuList[i][j]]);
        }
    }

    return 0;
}

除了上一题的注意点外，这一题还有几个值得关注的地方：
1. 对于学生的名字，仍然不能用string，否则最后一个测试点会超时。这里使用了一个二维数组来保存学生的输入顺序与学生姓名的映射关系，便于结果的输出，这一点挺巧妙的。
2. 对于输出的控制：要求按照学生姓名的字典序输出。这里的排序方式非常之巧妙！首先，没有直接对学生姓名字符串进行排序，而是根据其在stuList中的下标来排序，减少了复杂度；对应的，cmp()函数的设计也是很巧妙的，值得品味。