The Suspects--并查集

ACM专题学习三

题目

2019冠状病毒病（英语：Coronavirus disease 2019，缩写：COVID-19 ），是一种由严重急性呼吸系统综合症冠状病毒2型（缩写：SARS-CoV-2）引发的传染病。此病在全球各国大规模爆发并急速扩散，成为人类历史上致死人数最多的流行病之一。 很显然，目前最好的办法就是将所有可能的患者都隔离起来。 现在某高校正在排查可能的患者，这个高校中有多个社团，每个社团经常进行内部交流，一名学生可能会加入多个社团。学校认为一旦某个社团里出现一名可疑患者，这整个社团的学生都被视为是可能的患者。 现在请你帮忙找到这所学校的所有可能的患者。

输入

输入文件包含多组数据。

对于每组测试数据：

第一行为两个整数n和m, 其中n是学生的数量, m是社团的数量。0 < n <= 30000，0 <= m <= 500。

接下来m行，每一行有一个整数k，代表社团中学生的数量。之后，有k个整数代表这个社团里每个学生的编号（在0到n-1之间）。

n = m = 0表示输入结束，不需要处理。

输出

对于每组测试数据， 输出可能的患者数目。

样例输入

100 4
2 1 2
6 11 13 50 12 14
2 0 1
2 99 2
200 2
1 5
6 5 6 7 8 9 10
1 0
0 0

样例输出

4
1
1

思路

有一道相同的英文题有说0号学生是可疑患者，这道题竟然没有说...

0号学生是可疑患者，Ta所在的社团的其他成员都是可能的患者。要算出可能的患者的数目，我们可以先把第一个社团的成员连起来（取社团中第一位学生为父结点），接下来的社团里的成员也是各自联通（用父结点联通），如果社团成员还联通其他社团，那其他社团也会与这个社团联通，联通后父结点相同了，最后遍历所有学生，如果学生的父结点与0号学生的父结点相同，则sum++;

代码

#include<vector>
#include<iostream>
using namespace std;
int fa[30000],a[30000];
int find(int x)
{
    return fa[x]==x?x:fa[x]=find(fa[x]);
}
int main()
{
    int n,m;
    while(~scanf("%d%d",&n,&m))
    {
        if(n==0&&m==0)
            break;
        for(int i=0; i<n; i++)
            fa[i]=i;
        int sum=0;
        for(int i=0; i<m; i++)
        {
            int k,f1,f2;
            scanf("%d",&k);
            scanf("%d",&a[0]);
                 f1=find(a[0]);
            for(int j=1; j<k; j++)
            {
                scanf("%d",&a[j]);
                f2=find(a[j]);
                if(f1!=f2)
                    fa[f2]=f1;
            }
        }
        for(int i=0; i<n; i++)
            if(find(i)==fa[0])
                sum++;
        printf("%d\n",sum);
    }
    return 0;
}