并查集＿感染者 (并查集应用)

时间限制: 5 Sec  内存限制: 128 MB
提交: 311  解决: 141
[提交][状态][讨论版]

题目描述

2003年3月SARS病毒疯狂的席卷了全球。为了降低被感染的可能性，最好的办法就是将感染者隔离。
在NSYSU大学,有许多学生社团.同一个学生社团中的人会频繁的接触，而一个学生可能会加入很多个团体。为了防止SARS的传播，NSYSU搜集了所有学生社团的名单。
只要社团中有一个人被感染，那么社团中的每一个人都将被感染。可是，每当出现一个感染者，要确定每一个人是否被感染确实很困难的。你能否写一个程序找出所有的受感染的。

输入

输入包含多组数组。每组测试数据首先包含两个整数n和m，n（0 < n <= 30000）表示学生的数目，m（0 <= m <= 500）表示社团的数目。每个学生都有一个唯一的编号，编号取值为0到n-1，编号为0的学生是所有学生的中最初的唯一感染者。接下来有m个社团的名单，每个社团的名单在输入中为一行。每一行先输入一个数k表示社团总人数。接着是社团中k个成员的编号
当输入使 n = 0 且 m = 0 时，则输入结束。

输出

每组测试数据输出一个数占一行，表示被感染的总人数。

样例输入

100 4
2 1 2
5 10 13 11 12 14
2 0 1
2 99 2
200 2
1 5
5 1 2 3 4 5
1 0
0 0

样例输出

4
1
1

每个社团的所有人与该社团中的第一个人合并,同时统计人数,然后从编号0开始找根节点..

#include<bits/stdc++.h>
using namespace std;
struct node
{
    int pre;
    int num;
}a[30005];
int Find(int x)
{
    if(a[x].pre==x) return x;
    return a[x].pre=Find(a[x].pre);
}
void unit(int p,int q)
{
    p=Find(p);
    q=Find(q);
    if(p!=q) a[p].num+=a[q].num;
    a[q].pre=p;
}
int main()
{
    int n,m,k,p,q;
    while(scanf("%d%d",&n,&m)==2&&(n||m))
    {
        for(int i=0;i<n;i++)
            a[i].pre=i,a[i].num=1;
        while(m--)
        {
            scanf("%d%d",&p,&k);
            for(int i=1;i<p;i++)
                scanf("%d",&q),unit(k,q);
        }
        int ans=Find(0);
        printf("%d\n",a[ans].num);
    }
    return 0;
}