2020 GDUT評価コンテストIV F.ニュース配信

F.ニュース配信

リンク

タイトル説明
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");
	}
}

公開された24元の記事 ウォンの賞賛1 ビュー598

おすすめ

転載: blog.csdn.net/palax0/article/details/104828642