poj1611 - The Suspects

裸并查集

题意：n个学生，加入了m个团体，分别编号0到n-1，若一个人感染了，他所在的团队的所有人都会感染，要求多组数据感染的人数，规定0号已感染。

并查集思想：初始化，查，并；此题可以让每组的节点合并到第一个，并用另一个数组son[]统计这个团体中的人数。0所在的连通图的根节点的son[]就是结果

#include <iostream>

using namespace std;
int n,m,k,f,t,s[30005],son[30005];
int find(int x)
{//找到x的根节点
    if(s[x]!=x)
        return s[x]=find(s[x]);
    return x;
}
void un(int a,int b)
{//将这组其他的人合并到第一个人的根节点的连通图上
    int x=find(a);
    int y=find(b);
    if(x!=y)//根节点不相同，则合并，让y（自然b也是）的根节点只想x，且x的孩子增加；
    {s[y]=x;
    son[x]+=son[y];}

}
int main()
{
    while(cin>>n>>m&&n+m)
    {
        for(int i=0;i<n;i++)
        {
            s[i]=i;
            son[i]=1;
        }//初始化
        for(int i=0;i<m;i++)
        {
            cin>>k;
            if(k>=1)
            {
                cin>>f;
            }
            for(int j=1;j<k;j++)
            {
                cin>>t;
                un(f,t);
            } }
           int ans=son[find(0)];

        cout<<ans<<endl;
    }
    return 0;
}