题目要求如下:
可以看到有40000个学生和2500个课程,而要输出每个课程对应学生的名称。
这个问题非常简单,使用1个哈希表存储1个数组即可轻松解决这个问题。因此,利用C++中的map及vector头文件中的函数来解决这个问题。
下面是实现的代码:
#include <cstdio>
#include <vector>
#include <map>
#include <algorithm>
using namespace std;
int main() {
int K, N;
// K是课程数量,N是学生数量
while (scanf("%d %d", &N, &K) != EOF) {
//哈希表,使用字符串-整型数组
map<string,vector<int>> students;
int n, num;
char name[5] = {0};
for (int k = 0; k < K; ++k) {
scanf("%d %d", &n, &num);
for (int j = 0; j < num; ++j) {
//获取学生及选择课程的数量
scanf("%s", name);
// 查看学生名称是否存在,不存在则创建课程的列表
if (students.find(name) == students.end()) {
vector<int> courses;
courses.push_back(n);
students[name] = courses;
} else {
// 追加课程号
students[name].push_back(n);
}
}
}
for (int i = 0; i < N; ++i) {
scanf("%s",name);
int size = students[name].size();
printf("%s %d ",name, size);
if(size>1){
//排序课程号
sort(students[name].begin(),students[name].end());
}
for (int j = 0; j < size; ++j) {
printf("%d ", students[name][j]);
}
printf("\n");
}
}
return 0;
}
整个过程很简单,但是结果却时间超限,可以看到执行时间超过1s从而导致无法通过。
分析后发现可能因为是哈希表使用字符串作为键名导致执行速度缓慢的问题。因此,尝试使用将字符串转换为整型的方式来解决这个问题。
由于传入的学生名称由3个大写字母外加1个数字组成,因此对应的哈希值可以采用下面的方式。比如字符串XYZ1,对应的整型应该为2324251,可以将其分解为如下的过程:
X->23 Y->24 Z->25 XYZ->232425
整个哈希过程类似如下:
232425 = 230000 + 2400 + 25
由于学生数量有40000个,而哈希函数可以生成262626*10+10=175770个值,完全可以满足数量的使用。
下面是实现的代码:
#include <cstdio>
#include <vector>
#include <map>
#include <algorithm>
using namespace std;
int get_name_id(const char *name) {
//获取学生的ID
int n = 0;
for (int i = 0; i < 3; ++i) {
n = n * 100 + (name[i] - 'A');
}
n *= 10;
n += name[3] - '0';
return n;
}
int main() {
int K, N;
// K是课程数量,N是学生数量
while (scanf("%d %d", &N, &K) != EOF) {
int id;
map<int, vector<int>> students;
int n, num;
char name[5] = {0};
for (int k = 0; k < K; ++k) {
scanf("%d %d", &n, &num);
for (int j = 0; j < num; ++j) {
//获取学生及选择课程的数量
scanf("%s", name);
id = get_name_id(name);
if (students.find(id) == students.end()) {
vector<int> courses;
courses.push_back(n);
students[id] = courses;
} else {
students[id].push_back(n);
}
}
}
for (int i = 0; i < N; ++i) {
scanf("%s", name);
id = get_name_id(name);
vector<int> course = students[id];
int size = course.size();
printf("%s %d ", name, size);
if (size > 1) {
sort(course.begin(), course.end());
}
for (int j = 0; j < size; ++j) {
printf("%d ", course[j]);
}
printf("\n");
}
}
return 0;
}
最后是通过后的结果,差点就无法通过测试。