原文链接:我的个人博客
原题链接
PAT 1047 Student List for Course
考点
stl容器
思路
题目给定每一名学生所选的所有课程,要求输出每一门课程对应的学生姓名。
起初的想法是利用set<string> s[2600]
,来存放对应学生所选的课程,并且set容器默认情况下就是按字典顺序从小到大的。我本来以为这个想法已经很prefect了,没想到最后一个测试点还是超时了。感到无比悲伤!柳神的思路果然很清奇,姓名存放在name数组中,下标即对应id号。再向vector中push_back课程编号的学生id。然后再对vector进行排序输出。可能,,,尽量不用复杂的容器就不要用复杂的容器吧。
代码
最后一个测试点超时,只能拿到23分
#include <bits/stdc++.h>
using namespace std;
set<string> s[2600];
int N,K;
int main(){
scanf("%d%d",&N,&K);
for(int i=0;i<N;i++){
string stu;
int c;
cin>>stu>>c;
for(int j=0;j<c;j++){
int id;
scanf("%d",&id);
s[id].insert(stu);
}
}
for(int i=1;i<=K;i++){
cout<<i<<" "<<s[i].size()<<endl;
set<string>::iterator it = s[i].begin();
for(;it!=s[i].end();it++){
cout<<*it<<endl;
}
}
return 0;
}
柳神的代码
#include <iostream>
#include <vector>
#include <algorithm>
#include <string.h>
using namespace std;
char name[40010][5];
vector<int> course[2510];
bool cmp1(int a, int b) {
return strcmp(name[a], name[b]) < 0;
}
int main() {
int n, k;
scanf("%d %d", &n, &k);
for(int i = 0; i < n; i++) {
int cnt, temp;
scanf("%s %d", name[i], &cnt);
for(int j = 0; j < cnt; j++) {
scanf("%d", &temp);
course[temp].push_back(i);
}
}
for(int i = 1; i <= k; i++) {
printf("%d %d\n", i, course[i].size());
sort(course[i].begin(), course[i].end(), cmp1);
for(int j = 0; j < course[i].size(); j++)
printf("%s\n", name[course[i][j]]);
}
return 0;
}