【算法笔记】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 =  *  *  * ;
 const int N = ;
 vector<int> selectCourse[maxn];
 int getID(char name[]){
     int id = ;
     for(int i = ; i < ; i++){
         id = id *  + (name[i] - 'A');
     }
     id = id *  + name[] - '';
     return id;
 }
 int main(){
     char name[];
     int n, k;
     scanf("%d %d", &n, &k);
     for(int i = ; i < k; i++){
         int course, x;
         scanf("%d %d", &course, &x);
         for(int j = ; j < x; j++){
             scanf("%s", name);
             int id = getID(name);
             selectCourse[id].push_back(course);
         }
     }
     for(int i = ; 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 = ; j < selectCourse[id].size(); j++){
             printf(" %d", selectCourse[id][j]);
         }
         printf("\n");
     }
     return ;
 }

set版

#include<bits/stdc++.h>
using namespace std;
const int maxn =  *  *  * ;
const int N = ;
set<int> selectCourse[maxn];
int getID(char name[]){
    int id = ;
    for(int i = ; i < ; i++){
        id = id *  + (name[i] - 'A');
    }
    id = id *  + name[] - '';
    return id;
}
int main(){
    char name[];
    int n, k;
    scanf("%d %d", &n, &k);
    for(int i = ; i < k; i++){
        int course, x;
        scanf("%d %d", &course, &x);
        for(int j = ; j < x; j++){
            scanf("%s", name);
            int id = getID(name);
            selectCourse[id].insert(course);
        }
    }
    for(int i = ; 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("\n");
    }
    return ;
}