[code] PTA 胡凡算法笔记 DAY054

题目 A1039 Course List for Student

• 题意
  输入学生人数和课程数。之后输入课程编号及选课人数及所有学生姓名。最后输入所有学生姓名进行查询选课情况。输出格式为：学生姓名 课程数 各课程编号（递增顺序输出）

• 思路
  开始很暴力直接用的map做映射，结果最后有一个case爆了。看解析之后采用的是先把name进行哈希，然后只需要vector数组就可以完成本题。首先将学生姓名哈希为一个整型（范围为：26 * 26 * 26 * 10 + 1，三个字母一个数字），然后申请这么大的vector数组，在输入的时候按下表push课程号即可。在输出的时候再排序输出。

• Code in C++

#include <cstdio>
#include <cstring>
#include <vector>
#include <algorithm>

const int M = 26 * 26 * 26 * 10 + 1;
const int N = 40001;
char name[5];
std::vector<int> tables[M];

int getID(char *name)
{
    int result = 0;
    for (int i = 0; i < 3; ++i)
    {
        result = result * 26 + name[i] - 'A';
    }
    result = result * 10 + name[3] - '0';
    return result;
}

int main()
{
    int n, k, index, num;
    scanf("%d %d", &n, &k);
    for (int i = 0; i < k; ++i)
    {
        scanf("%d %d", &index, &num);
        for(int j = 0; j < num; ++j)
        {
            scanf("%s", name);
            int id = getID(name);
            tables[id].push_back(index);
        }
    }

    // 查询
    for (int i = 0; i < n; ++i)
    {
        scanf("%s", name);
        int id = getID(name);
        std::sort(tables[id].begin(), tables[id].end());
        int counter = tables[id].size();
        printf("%s %d", name, counter);
        for(int j = 0; j < counter; ++j)
        {
            printf(" %d", tables[id][j]);
        }
        printf("\n");
    }
    return 0;
}

---

小结

字符串哈希为int来降低实现复杂度。