POJ 1611 The Suspects(并查集)

POJ 1611 The Suspects(并查集)

http://poj.org/problem?id=1611

题意：

        现在有n个学生(从0号到n-1号),其中0号学生是有可能非典的,只要和被怀疑有非典的学生在一个社团的学生都是有可能有非典的且需要被隔离,但是学校有很多社团,所以现在要你求一共有多少学生需要被隔离?

        输入:多组实例.每个实例第一行n和m (0 <n <= 30000 and 0 <= m <= 500),表示学生从0到n-1号,接下来有m行,每行描述一个社团的所有成员.每行首先是一个k,表示该社团有k个成员,接下来是这k个成员的编号.当一组实例n=0且m=0时,表示输入结束.

        输出:需要被隔离的学生数.

分析:

        所有学生被处理前都是独立的连通分量，以自己作为根。只要读入一个社团，那么社团中所有的学生所属的连通分量都要合并成1个。不断这么处理，最终和0号学生在同一个连通分量的所有学生都是需要被隔离的。

        AC代码(新)中用 根节点维护num[i]信息表示当前连通分量的节点数目，当所有节点合并完之后，只需要计算0号节点所在的连通分量的点数 输出即可。

AC代码(新):

#include<cstdio>
#include<cstring>
#include<vector>
using namespace std;
const int maxn=30000+5;

int fa[maxn];
int num[maxn];//根节点数目
int findset(int x)
{
    return fa[x]==-1? x:fa[x]=findset(fa[x]);
}
void bind(int u,int v)
{
    int fu=findset(u);
    int fv=findset(v);
    if(fu!=fv)
    {
        num[fv] += num[fu];
        fa[fu] = fv;
    }
}

int main()
{
    int n,m;
    while(scanf("%d%d",&n,&m)==2 && n)
    {
        for(int i=0;i<n;i++)
        {
            fa[i]=-1;
            num[i]=1;
        }

        while(m--)
        {
            int k, u, v;
            scanf("%d%d",&k,&u);
            for(int i=1;i<=k-1;i++)
            {
                scanf("%d",&v);
                bind(u,v);
            }
        }

        printf("%d\n",num[findset(0)] );
    }
    return 0;
}

AC代码:16ms

#include<cstdio>
#include<cstring>
using namespace std;
const int MAXN=30000+100;
int pa[MAXN];
int findset(int x)
{
    if(pa[x]==-1)return x;
    return pa[x]=findset(pa[x]);
}
void bind(int i,int j)
{
    int fa=findset(i);
    int fb=findset(j);
    if(fa!=fb)
        pa[fb]=fa;
}
int main()
{
    int n,m;
    while(scanf("%d%d",&n,&m)==2)
    {
        int ans=1;
        if(n==0&&m==0)break;
        memset(pa,-1,sizeof(pa));
        while(m--)//读m组的成员
        {
            int k,first;
            scanf("%d",&k);
            if(k==0)continue;//以防k为0,即该组没人
            scanf("%d",&first);//读该组的第一个成员
            k--;
            while(k--)
            {
                int a;
                scanf("%d",&a);
                bind(first,a);
            }
        }
        int g=findset(0);
        for(int i=1;i<n;i++)
        {
            if(g == findset(i))
                ans++;
        }
        printf("%d\n",ans);
    }
    return 0;
}