[code] PTA 胡凡算法笔记 DAY055

题目 A1047 Student List for Course

• 题意
  输入学生人数和课程数。输入学生名字，选择课程数和各选择课程编号。输出每门课程选课的学生数。课程编号从小到大输出，学生名字按字母序输出。

• 思路
  根据需要输出的内容很明显可以看出这里映射课程号是key，学生姓名是value。而且因为同一个学生会选择很多课程，所以名字会存在大量的冗余。所以为了减少开销，这里用一个数组建立一个char* -> int的哈希表，降低一部分开销。课程的映射表则采用下标和vector<int>来表示。
  这里哈希算法getID()还是跟上一题一样，像计算数转换为ASCII进行。输出要求部分课程编号递增直接遍历就好，字母序因为这个哈希算法的特征，直接对应从小到大的顺序排序输出即可。

• Code in C++

#include <cstdio>
#include <vector>
#include <cstring>
#include <algorithm>
const int M = 26 * 26 * 26 * 10 + 1;
const int N = 2501;
// 哈希表
char names[M][5];
std::vector<int> course[N];

// 获取姓名的哈希值
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;
}


// 输出结果
void print(int k)
{
    for (int i = 1; i <= k; ++i)
    {
        int lenth = course[i].size();
        printf("%d %d\n", i, lenth);
        std::sort(course[i].begin(), course[i].end());  // 字母序即从小到大排序
        for (int j = 0; j < lenth; ++j)
        {
            printf("%s\n", names[course[i][j]]);
        }
    }
}

int main()
{
    int n, k;
    char name[5];
    scanf("%d %d", &n, &k);
    for (int i = 0; i < n; ++i)
    {
        int m, tmp;
        scanf("%s", name);
        int id = getID(name);
        strcpy(names[id], name);
        scanf("%d", &m);
        for (int j = 0; j < m; ++j)
        {
            scanf("%d", &tmp);
            course[tmp].push_back(id);
        }
    }

    // 输出
    print(k);
    return 0;
}

---

小结

• 抽象出需要存储的数据结构，然后通过哈希的方式帮助降低冗余。
• 数据范围很大的情况，一般最好用char数组存储，用string很容易超时。