PAT A1039 Course List for Student (25 分)

题目注意点

• 数据太大，要用字符串hash把姓名和数字一一对应
• 要用vector减少空间消耗
• 输出学生所选课程时要按递增序输出

实现

每个学生选择的课程都存在一个vector里面，因此需要创建一个vector的数组存放所有学生的选课情况
算法笔记题解：

#include <stdio.h>
#include <algorithm>
#include <vector>
using namespace std;

const int M = 26 * 26 * 26 * 10 + 1;
vector<int> selectCourse[M];//每个学生选择的课程编号

int getID(char name[]) {//hash函数，将字符串name转换成数字
    int id = 0;
    for (int i = 0; i < 3; i++) {
        id = id * 26 + (name[i] - 'A');
    }
    id = id * 10 + (name[3] - '0');
    return id;
}

int main() {
    int n, k;
    char name[5];
    scanf("%d%d", &n, &k);
    for (int i = 0; i < k; i++) {
        int index, stuNum;
        scanf("%d%d", &index, &stuNum);
        for (int j = 0; j < stuNum; j++) {
            scanf("%s", name);
            int id = getID(name);
            selectCourse[id].push_back(index);
        }
    }
    for (int i = 0; i < n; i++) {
        scanf("%s", name);
        int id = getID(name);
        sort(selectCourse[id].begin(), selectCourse[id].end());
        int size = selectCourse[id].size();
        printf("%s %d", name, size);
        for (int j = 0; j < size; j++) {
            printf(" %d", selectCourse[id][j]);
        }
        printf("\n");
    }
    return 0;
}