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;
}