2020 GDUT Rating Contest IV F. News Distribution

F. News Distribution

链接

题目描述
有n个人，他们各加入了一些群。给出m条关于某个群有谁的信息。若一个人的群友加入了另一个群，这个人也算加入了另一个群。输出每个人加入的群数（给出信息之前默认每个人已经加入一个群）

题目分析
因为是人与人直接有传递性的关系，套用并查集模板即可。

代码

#include <bits/stdc++.h>

using namespace std;

int root[500001],ans[500001];

int union_search(int x){
	int son=x,temp;
	while(root[x]!=x){
		x=root[x];
	}
	while(root[son]!=x){
		temp=son;
		son=root[son];
		root[temp]=x;
	}
	return x;
}

int main(){
	int n,m;
	cin>>n>>m;
	for(int i=1;i<=n;i++)
		root[i]=i;
	for(int i=0;i<m;i++){
		int member,first;
		scanf("%d",&member);
		if(member==0)
			continue;
		scanf("%d",&first);
		int root_first=union_search(first);
		for(int j=0;j<member-1;j++){
			int temp;
			scanf("%d",&temp);
			temp=union_search(temp);
			if(temp!=root_first){
				root[temp]=root_first;//是temp的根等于root_first而不是temp
				//printf("now,%d's father is %d\n",temp,root_first);
			}
		}
	}
	for(int i=1;i<=n;i++){
		union_search(i);
	}
	for(int i=1;i<=n;i++){
		ans[root[i]]++;
	}
	for(int i=1;i<=n;i++){
		//printf("%d's father is %d\n",i,root[i]);
		printf("%d",ans[root[i]]);
		if(i!=n)
			printf(" ");
		else
			printf("\n");
	}
}