STL_A1039 Course List for Student (25 分)

A1039 Course List for Student (25 分)

/*
*vector的使用
*使用map、string容易超时
*使用字符串hash把name转换为number
*/

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

const int M=26*26*26*10+1;
vector<int> selectCourse[M];

int nameToNum(char name[]) { //把字符串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 course,x;
        scanf("%d%d",&course,&x);
        for(int j=0;j<x;j++) {
            scanf("%s",name);
            int id=nameToNum(name);
            selectCourse[id].push_back(course);
        }
    }
    for(int i=0;i<n;i++) {
        scanf("%s",name);
        int id=nameToNum(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("\n");
    }
    return 0;
}