2020 GDUT Rating Contest IV F. News Distribution

最新推荐文章于 2020-03-13 02:09:12 发布

palax0

最新推荐文章于 2020-03-13 02:09:12 发布

阅读量456

点赞数

分类专栏：排位赛

本文链接：https://blog.csdn.net/palax0/article/details/104828642

版权

排位赛专栏收录该内容

15 篇文章 0 订阅

订阅专栏

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++){//保证他们与root直接相连
		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");
	}
}