PAT A1039 Course List for Student (25 分) 排序

题目大意：给出K门课程的学生信息，查找指定的N名学生的选课信息。

用map对应即可，不过虽然能过，时间消耗比较大。

#include <bits/stdc++.h>

using namespace std;

map<string, vector<int>> mp;

int main()
{
    int N, K;
    cin >> N >> K;
    for (int i = 0; i < K; ++i)
    {
        int courseId, stuNum;
        cin >> courseId >> stuNum;
        for (int j = 0; j < stuNum; ++j)
        {
            string name;
            cin >> name;
            mp[name].push_back(courseId);
        }
    }
    for (int query = 0; query < N; ++query)
    {
        string name;
        cin >> name;
        printf("%s %zu", name.c_str(), mp[name].size());
        vector<int> course = mp[name];
        sort(course.begin(), course.end());
        for (int i = 0; i < course.size(); ++i)
        {
           printf(" %d", course[i]);
        }
        printf("\n");
    }
    return 0;
}

优化一下：

① 用scanf 输入，那么姓名的储存需要用char数组。

② 将姓名字符串对应到数字id。由于姓名字符串是固定的三位大写字母+一位数字的形式，最简单的对应方式是将这个字符串看成是一个4位的36进制数，然后对应即可。

③ map用数字id作为键。

#include <bits/stdc++.h>

using namespace std;

map<int, vector<int>> mp;

int getId(char* name)
{
    return (name[0] - 'A') * pow(36,3) + (name[1] - 'A') * pow(36,2) + (name[2] - 'A') * 36 + name[3] - '0';
}

int main()
{
    int N, K;
    scanf("%d%d", &N, &K);
    for (int i = 0; i < K; ++i)
    {
        int courseId, stuNum;
        cin >> courseId >> stuNum;
        for (int j = 0; j < stuNum; ++j)
        {
            char name[4];
            scanf("%s", name);
            mp[getId(name)].push_back(courseId);
        }
    }
    for (int query = 0; query < N; ++query)
    {
        char name[4];
        scanf("%s", name);
        int id = getId(name);
        printf("%s %zu", name, mp[id].size());
        sort(mp[id].begin(), mp[id].end());
        for (int i = 0; i < mp[id].size(); ++i)
        {
           printf(" %d", mp[id][i]);
        }
        printf("\n");
    }
    return 0;
}