7-13 社交集群 (30 分)
当你在社交网络平台注册时,一般总是被要求填写你的个人兴趣爱好,以便找到具有相同兴趣爱好的潜在的朋友。一个“社交集群”是指部分兴趣爱好相同的人的集合。你需要找出所有的社交集群。
输入格式:
输入在第一行给出一个正整数 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
并查集思想,把每一种出现过的爱好标记,一旦出现标记过的爱好,就把当前编号和前一个编号合并,最后统计并查集的个数
#include<bits/stdc++.h>
using namespace std;
int a[100010];
int p[10010];
int find(int x){
if(p[x]!=x)p[x]=find(p[x]);
return p[x];
}
vector<int>v[10010];
map<int,int>mp;
int main(){
int n;
cin>>n;
for(int i=1;i<=n;i++){
p[i]=i;
}
for(int i=1;i<=n;i++){
int m;
scanf("%d:",&m);
while(m--){
int x;
cin>>x;
v[i].push_back(x);
if(mp[x]==0){mp[x]=i;}
else{
p[find(i)]=find(mp[x]);
}
}
}
int k=0,k1=0;
for(int i=1;i<=n;i++){
if(p[i]==i)k++;
a[find(i)]++;
}
cout<<k<<endl;
sort(a+1,a+1+n);
for(int i=n;i>0;i--){
if(a[i]!=0){
if(k1++)cout<<' ';
cout<<a[i];
}
}
}