当你在社交网络平台注册时,一般总是被要求填写你的个人兴趣爱好,以便找到具有相同兴趣爱好的潜在的朋友。一个“社交集群”是指部分兴趣爱好相同的人的集合。你需要找出所有的社交集群。
输入格式:
输入在第一行给出一个正整数 N(≤1000),为社交网络平台注册的所有用户的人数。于是这些人从 1 到 N 编号。随后 N 行,每行按以下格式给出一个人的兴趣爱好列表:
Ki: hi[1] hi[2] ... hi[Ki]
其中Ki(>0)是兴趣爱好的个数,hi[j]是第j个兴趣爱好的编号,为区间 [1, 1000] 内的整数。
输出格式:
首先在一行中输出不同的社交集群的个数。随后第二行按非增序输出每个集群中的人数。数字间以一个空格分隔,行末不得有多余空格。
输入样例:
8
3: 2 7 10
1: 4
2: 5 3
1: 4
1: 3
1: 4
4: 6 8 1 5
1: 4
输出样例:
3
4 3 1
解题思路:
这是一个简单的并查集问题,但是在找编号i的父亲的时候如果直接模拟的去找很可能会超时,因为最坏情况下输入是的时间,再把第i个人和前面的i-1个人比较是否有相同项的时候又会是
.
在这里我用一个数组f处理了一下可以O(1)的查找第i个人的父亲是谁,把兴趣j作为下标,f[j]表示有兴趣j的第一个人的编号,这样我们输入的时候比如说输入a,直接可以把第i个人的父亲设置为f[a];
AC代码:
#include<bits/stdc++.h>
using namespace std;
int n,k;
char c;
int fa[1003];
int f[1003];
map<int,int>vis;
bool cmp(int a,int b){
return a>b;
}
void init(){
for(int i=1;i<=n;i++)fa[i]=i;return;
}
int find(int x){
if(fa[x]==x)return x;
else return fa[x]=find(fa[x]);
}
void m(int x,int y){
x=find(x);
y=find(y);
fa[x]=y;
return;
}
int main(){
cin >>n;
init();
cin >>k>>c;
for(int i=1;i<=k;i++){
int a;
cin >>a;
f[a]=1;
}
for(int i=2;i<=n;i++){
cin >>k>>c;
for(int j=1;j<=k;j++){
int a;
cin >>a;
if(f[a]==0)f[a]=i;
else if(f[a]==i)continue;
else{
m(i,f[a]);
}
}
}
int res=0;
for(int i=1;i<=n;i++){
int xx=find(i);
if(vis[xx]==0)res++;
vis[xx]++;
}
cout<<res<<endl;
vector<int>a;
for(map<int,int>::iterator it=vis.begin();it!=vis.end();it++){
a.push_back(it->second);
}
sort(a.begin(),a.end(),cmp);
for(int i=0;i<res;i++){
cout<<a[i];
if(i!=res-1)cout<<" ";
}
return 0;
}