L3-003. 社交集群_并查集_2018_2_22

L3-003. 社交集群

时间限制

1000 ms

内存限制

65536 kB

代码长度限制

8000 B

判题程序

Standard

作者

陈越

在社交网络平台注册时，用户通常会输入自己的兴趣爱好，以便找到和自己兴趣相投的朋友。有部分兴趣相同的人们就形成了“社交集群”。现请你编写程序，找出所有的集群。

输入格式：

输入的第一行给出正整数N（<=1000），即社交网络中的用户总数（则用户从1到N编号）。随后N行，每行按下列格式列出每个人的兴趣爱好：

K_i: h_i[1] h_i[2] ... h_i[K_i]

其中K_i（>0）是第i个人的兴趣的数量，h_i[j]是第i个人的第j项兴趣的编号，编号范围为[1, 1000]内的整数。

输出格式：

首先在第一行输出整个网络中集群的数量，然后在第二行按非递增的顺序输出每个集群中用户的数量。数字间以1个空格分隔，行首尾不得有多余空格。

输入样例：

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<iostream>
#include<cstring>
#include<set>
#include<algorithm>
#include<vector> 
using namespace std;

const int N=1e3+10;

int fa[N];
int ans[N];
typedef set<int> si;
typedef vector<int> vi;

int parent(int x){
	if(fa[x]==-1)return x;
	return fa[x]=parent(fa[x]);
}

int main(){
	int n;
	scanf("%d",&n);
	int m;
	memset(fa,-1,sizeof(fa));
	vi v(n);
	for(int i=0;i<n;i++){
		scanf("%d:",&m);
		int p,q;
		scanf("%d",&p);
		v[i]=p;
		while(--m){
			scanf("%d",&q);
			p=parent(p);
			q=parent(q);
			if(p!=q)fa[p]=q;
		}
	}
	si s;int x;
	for(int i=0;i<n;i++){
		x=parent(v[i]);
		s.insert(x);
		ans[x]++;
	}
	v.clear();
	for(si::iterator it=s.begin();it!=s.end();it++)
	v.push_back(ans[*it]);
	
	sort(v.begin(),v.end(),greater<int>());
	x=v.size();
	printf("%d\n",x);
	int i;
	for(i=0;i<x-1;i++)
	printf("%d ",v[i]);
	printf("%d\n",v[i]);
}