POJ1611 The Suspects 并查集

原创于 2014-12-31 11:20:22 发布 · 445 阅读

0 ·

CC 4.0 BY-SA版权

图论专栏收录该内容

58 篇文章

订阅专栏

本文介绍了一个使用并查集算法解决的问题：如何确定受SARA病毒感染的学生需要隔离的人数。通过构建并查集，算法能够高效地找出与初始感染者处于同一组的所有学生。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

题目大意：有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;
}