POJ1611 The Suspects 并查集

题目大意:有n个学生,分成m组,SARA病毒感染编号为0的学生,某一组学生要被隔离,当且仅当该组里有至少一个人感染SARA病毒,现在问你一共需要隔离多少学生。

典型的并查集问题,找出编号为0的学生所在的集合有多少元素即可。

#include <cstdio>
#include <iostream>
using namespace std;
#define MAX 30010
int par[MAX],rank[MAX];
void Init(int n)
{
    for(int i=0;i<n;i++)
    {
        par[i]=i;
        rank[i]=1;
    }
}
int Find(int x)
{
    if(par[x]!=x)
      return par[x]=Find(par[x]);
    return par[x];
}
void Union(int x,int y)
{
    x=Find(x);
    y=Find(y);
    if(x==y)  return ;
    if(rank[x]>rank[y])
    {
        par[y]=x;
        rank[x]+=rank[y];
    }
    else
    {
        par[x]=y;
        rank[y]+=rank[x];
    }
}
int main()
{
    int n,m,i,k;
    int ans,tmp;
    while(scanf("%d%d",&n,&m))
    {
        if(m==0&&n==0) break;
        Init(n);
        while(m--)
        {
            scanf("%d%d",&k,&ans);
            for(i=1;i<k;i++)
            {
                scanf("%d",&tmp);
                Union(ans,tmp);
            }
        }
        printf("%d\n",rank[Find(0)]);
    }
    return 0;
}


 

你可能感兴趣的:(POJ1611 The Suspects 并查集)