POJ-The Suspects(并查集模板题)

加油。
To be better.

题目链接：
https://vjudge.net/problem/POJ-1611

题目大意：
给你一个n和m，n代表学生总数，m代表团体数，接下来m行，每一行输入一个t代表团体中的学生数，接下来输入t个学生编号代表在这个团体内。注意一旦团体中有一个SARS病毒可能感染者，那么整个团体就都是SARS病毒可能的感染者。默认0号是SARS病毒可能的感染者。

题解：
这其实是一个并查集模板题，但是先开始我卡在了怎么样让所有的感染者下标都指向0，实际上，并不用这样做，只需要常规的合并集合，最后从1到n遍历一下谁和0是一个祖先，即Find（x）==Find（0）；

AC代码：

#include<iostream>
using namespace std;
struct node
{//我这里用的是结构题，方便路径压缩
    int x,pre,rak;
} f[30011];
int n,m;

void Init()
{//初始化
    for(int i=0; i<=n; i++)
    {
        f[i].pre=i;
        f[i].rak=1;
    }
}

int Find(int x)
{//查找函数
    if(f[x].pre==x)
        return x;
    else
        return Find(f[x].pre);
}

void Union(int x,int y)
{//合并（路径压缩优化）
    int rx=Find(f[x].pre);
    int ry=Find(f[y].pre);
    if(rx!=ry)
    {
        if(f[rx].rak<f[ry].rak)
            f[rx].pre=f[ry].pre;
        else
        {
            f[ry].pre=f[rx].pre;
            if(f[rx].rak==f[ry].rak)
                f[rx].rak++;
        }
    }
}

int Query()
{//查询一共有多少个可能的感染者
    int t=Find(f[0].pre);
    int ans=1;
    for(int i=1;i<=n;i++)
    {
        if(t==Find(f[i].pre))
            ans++;
    }
    return ans;
}
int main()
{
    int t,a,b;
    while(cin>>n>>m)
    {
        if(n==0)
            break;
        if(m==0)
            cout<<1<<endl;
        else
        {
            Init();
            for(int i=1; i<=m; i++)
            {
                cin>>t;
                for(int j=1; j<=t; j++)
                {
                    if(j==1)
                        cin>>a;//第一个数单独输入，方便后续合并
                    else
                    {
                        cin>>b;
                            Union(a,b);
                    }
                }
            }
            cout<<Query()<<endl;
        }
    }
    return 0;
}