Week6 作业 B

题目描述：

有n个学生，m个学生群体，其中有一个编号为0的学生感染了病毒，则只要是与0号学生直接接触或传递接触的人，都是The Suspects疑似病人，问有多少个疑似病人。

思路：

并查集。输入时，可以单独输入每个学生团体的第一个人，然后让后边的人都与第一个人进行合并。

代码：

（未使用重量规则）

#include <cstdio>
#include <iostream> 
const int MAXN=1e5+5;
using namespace std;
int fa[MAXN];
void init(int n)
{
    for(int i=0;i<n;i++)
        fa[i]=i;
}
int find(int x)
{
    if(fa[x]==x) return fa[x];
    fa[x]=find( fa[x] );
    return fa[x];
}
void unite(int x,int y)
{
    int root1=find(x),root2=find(y);
    fa[root1]=root2;
}
int main()
{
    int n,m;
    while( scanf("%d %d",&n,&m) && !(m==0 && n==0))
    {    
        init(n);
        for(int i=1;i<=m;i++)
        {
            
            int num,t2,t3; scanf("%d %d",&num,&t2);
            for(int j=2;j<=num;j++)
            {
                scanf("%d",&t3);
                unite(t2,t3);
            }
        }
        int ans=0;
        for(int i=0;i<n;i++)
        {
            if( find(i)==find(0) ) ans++;
        }
        cout<<ans<<endl;
    }    
}