1039. 学生的课程列表
思路:
思路并不难,但有两点要注意:
- 相比string,使用char数组更快,本题中使用string会运行超时;
- 使用了哈希表,即将char数组表示的name映射成对应的整型数据,这样可以进一步加快查找速度。
本题中,如何通过上面两步缩短运行时间是考察的重点。
#include<iostream>
#include<vector>
#include<set>
#include<map>
#include<queue>
#include<algorithm>
#include<string>
#include<string.h>
using namespace std;
vector<int> name2course[26*26*26*10];
int main() {
int n, k;
scanf("%d%d",&n,&k);
vector<vector<char*>> course(k+1);
for(int i = 0; i < k; ++i) {
int cid, num;
scanf("%d%d",&cid,&num);
course[cid].resize(num);
for(int j = 0; j < num; ++j)
{
char* name = new char[4];
scanf("%s",name);
course[cid][j] = name;
}
}
for(int i = 1; i <= k; ++i) {
for(int j = 0; j < course[i].size(); ++j)
{
char* name = course[i][j];
int index = (name[0]-'A')*26*26*10+(name[1]-'A')*26*10+(name[2]-'A')*10+(name[3]-'0');
name2course[index].push_back(i);
}
}
//for query
for(int i = 0; i < n; ++i)
{
char name[4];
scanf("%s",name);
printf("%s",name);
int index = (name[0]-'A')*26*26*10+(name[1]-'A')*26*10+(name[2]-'A')*10+(name[3]-'0');
printf(" %d", name2course[index].size());
for(int j = 0; j < name2course[index].size(); ++j)
{
printf(" %d", name2course[index][j]);
}
printf("\n");
}
return 0;
}