[code] PTA 胡凡算法笔记 DAY054

Posted 2023-02-07 wait_for_that_day5

文章目录

• • 题目 A1039 Course List for Student
  • 小结

题目 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来降低实现复杂度。