POJ 1611 The Suspects（并查集）

Description

有n名学生（编号0~n-1）分成了m组，其中学生0感染了病毒，这种病毒极易传播，两人只要接触就会传播，一人感染全组感染，问共有多少名学生会感染这种病毒？

Input

多组用例，每组用例第一行为两个整数n和m表示学生人数和组数，之后m行每行首先输入该组学生人数num，然后输入num个整数表示该组学生的编号，以 0 0结束输入

Output

对于每组用例，输出被感染学生的数量

Sample Input

100 4

2 1 2

5 10 13 11 12 14

2 0 1

2 99 2

200 2

1 5

5 1 2 3 4 5

1 0

0 0

Sample Output

4

1

1

Solution

并查集，用num[i]表示每个集合的元素数量，在合并过程中加上num的累加过程，最后答案即为num[find(0)]

Code

#include<cstdio>
#include<iostream>
#include<cstring>
using namespace std;
#define maxn 33333
int fa[maxn];
int deep[maxn];
int num[maxn];
void init(int n)
{
for(int i=0;i<=n;i++)
{
fa[i]=i;
deep[i]=0;
num[i]=1;
}
}
int find(int x)
{
if(fa[x]==x) return x;
else return fa[x]=find(fa[x]);
}
void unite(int x,int y)
{
x=find(x);
y=find(y);
if(x==y) return;
if(deep[x]<deep[y]) fa[x]=y,num[y]+=num[x];
else
{
fa[y]=x,num[x]+=num[y];
if(deep[x]==deep[y]) deep[x]++;
}
}
int main()
{
int n,m;
while(~scanf("%d%d",&n,&m),n||m)
{
init(n);
for(int i=0;i<m;i++)
{
int cnt,pre,last;
scanf("%d",&cnt);
if(cnt)scanf("%d",&pre);
if(cnt>1)
for(int j=1;j<cnt;j++)
{
scanf("%d",&last);
unite(pre,last);
pre=last;
}
}
printf("%d\n",num[find(0)]);
}
return 0;
}