版本2更新
不超时版
#include <string>
#include <vector>
#include <set>
#include <unordered_map>
#include <iostream>
#include <algorithm>
using namespace std;
int main(){
int n, k, c, x;
cin>>n>>k;
vector<vector<string>> V(k+1);
string name;
while(n--){
cin>>name>>c;
while(c--){
scanf("%d", &x);
V[x].push_back(name);
}
}
for(int i = 1; i <= k; i++){
sort(V[i].begin(), V[i].end());
printf("%d %d\n",i, V[i].size());
for(auto p: V[i]) printf("%s\n", p.c_str());
}
return 0;
}
版本2
牛客网测试点
以前写的代码多稚嫩,果然还是要多刷题呀!
在PTA上面有点超时
#include <string>
#include <vector>
#include <set>
#include <unordered_map>
#include <iostream>
#include <algorithm>
using namespace std;
const int maxn = 307;
int main(){
int n, k, c, x;
cin>>n>>k;
vector<vector<string>> V(k+1);
string name;
while(n--){
cin>>name>>c;
while(c--){
cin>>x;
V[x].push_back(name);
}
}
for(int i = 1; i <= k; i++){
sort(V[i].begin(), V[i].end());
cout<<i<<" "<<V[i].size()<<endl;
for(auto p: V[i]) cout<<p<<endl;
}
return 0;
}
版本1
第一次提交用的map映射,最后有个测试点超时,我以为时cout输出导致的,后来想了一下才发现有更简单的方法做出来,所以这个超时是必然的。
这里直接用vector记录名字就完成了。
再sort函数中,如果不自定义比较方式,通常是按照字符串的字典序排列的。
#include<iostream>
#include<cstdio>
#include<set>
#include<string>
#include<cstring>
#include<map>
#include<vector>
#include<algorithm>
using namespace std;
const int maxn = 4e4+5;
vector<string> v[maxn];
int main(){
int n,m,k,tmp;
string name;
scanf("%d%d",&n,&m);
for(int i=0;i<n;i++){
cin>>name>>k;
while(k--){
cin>>tmp;
v[tmp].push_back(name);
}
}
for(int i=1;i<=m;i++){
sort(v[i].begin(),v[i].end());
}
for(int i=1;i<=m;i++){
printf("%d %d\n",i,v[i].size());
for(auto it: v[i]){
printf("%s\n",it.c_str());
}
}
return 0;
}