【算法笔记】A1039 Course List for Student

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

题意：

　　有N个学生，K节课。给出选择每门课的学生姓名，，并在之后给出这N个学生姓名，按顺序输出每个学生选了几门课、哪几门课。

思路：

 　　用hash表存储每个学生的选课情况，需要定义一个hash函数把姓名转换为数字。查询时先把课程用sort()排序再输出。

　　最后一个测试点408ms，把 string name 改成 char name[] 之后106ms，说明大部分情况下string比较耗时。个别情况下char型数组会比较耗时。传送门

　　可以用vector数组保存选课，也可以用set数组来保存，vector数组需要对课程排序，而set数组插入时自动排序但是只能通过迭代器访问（只有vector和string能使用下标或*(it + i)访问），不过对本题来说没有影响。

vector版

#include<bits/stdc++.h>
using namespace std;
const int maxn = 26 * 26 * 26 * 10;
const int N = 40010;
vector<int> selectCourse[maxn];
int getID(char 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(){
    char name[5];
    int n, k;
    scanf("%d %d", &n, &k);
    for(int i = 0; i < k; i++){
        int course, x;
        scanf("%d %d", &course, &x);
        for(int j = 0; j < x; j++){
            scanf("%s", name);
            int id = getID(name);
            selectCourse[id].push_back(course);
        }
    }
    for(int i = 0; i < n; i++){
        scanf("%s", name);
        int id = getID(name);
        sort(selectCourse[id].begin(), selectCourse[id].end());
        printf("%s %d", name, selectCourse[id].size());
        for(int j = 0; j < selectCourse[id].size(); j++){
            printf(" %d", selectCourse[id][j]);
        }
        printf("
");
    }
    return 0;
}

 set版

#include<bits/stdc++.h>
using namespace std;
const int maxn = 26 * 26 * 26 * 10;
const int N = 40010;
set<int> selectCourse[maxn];
int getID(char 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(){
    char name[5];
    int n, k;
    scanf("%d %d", &n, &k);
    for(int i = 0; i < k; i++){
        int course, x;
        scanf("%d %d", &course, &x);
        for(int j = 0; j < x; j++){
            scanf("%s", name);
            int id = getID(name);
            selectCourse[id].insert(course);
        }
    }
    for(int i = 0; i < n; i++){
        scanf("%s", name);
        int id = getID(name);
        printf("%s %d", name, selectCourse[id].size());
        set<int>::iterator it = selectCourse[id].begin();
        for(; it != selectCourse[id].end(); it++){
            printf(" %d", *it);
        }
        printf("
");
    }
    return 0;
}