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");
}
}
标签:Rating,GDUT,temp,Contest,int,加入,son,500001,root 来源: https://blog.csdn.net/palax0/article/details/104828642