ACM 之 B - The Suspects

Description

严重急性呼吸系统综合症( SARS), 一种原因不明的非典型性肺炎,从2003年3月中旬开始被认为是全球威胁。为了减少传播给别人的机会, 最好的策略是隔离可能的患者。
在Not-Spreading-Your-Sickness大学( NSYSU), 有许多学生团体。同一组的学生经常彼此相通,一个学生可以同时加入几个小组。为了防止非典的传播,NSYSU收集了所有学生团体的成员名单。他们的标准操作程序(SOP)如下：
一旦一组中有一个可能的患者, 组内的所有成员就都是可能的患者。
然而,他们发现当一个学生被确认为可能的患者后不容易识别所有可能的患者。你的工作是编写一个程序, 发现所有可能的患者。

Input

输入文件包含多组数据。
对于每组测试数据：
第一行为两个整数n和m, 其中n是学生的数量, m是团体的数量。0 < n <= 30000，0 <= m <= 500。
每个学生编号是一个0到n-1之间的整数，一开始只有0号学生被视为可能的患者。
紧随其后的是团体的成员列表，每组一行。
每一行有一个整数k，代表成员数量。之后,有k个整数代表这个群体的学生。一行中的所有整数由至少一个空格隔开。
n = m = 0表示输入结束，不需要处理。

Output

对于每组测试数据, 输出一行可能的患者。

Sample Input

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

Sample Output

4
1
1

理解

考的是并查集, 不知道的同学可以去查一下,学会并查集之后这一题应该非常简单.

代码部分

#include<iostream>
using namespace std;
int m,n,x,sum,parent[30001],a[30001];
void set()
{
    for(int i=0;i<=m;i++)
        parent[i]=i;
}//最初每个数字都单独一个结点,没有父类没有子类.
int find(int a)
{
    if(parent[a]==a)
        return a;
    else
        return parent[a]=find(parent[a]);
}//寻找父点(即根节点)
void Union(int a,int b)
{
    int t1=find(a);
    int t2=find(b);
    if(t1!=t2)
        parent[t2]=t1;
}//将输入的一串数字组合成一条边,附上唯一根节点.
int main()
{
    while(cin>>m>>n)
    {
        if(m==0&&n==0)
            return 0;
        set();
        for(int i=0;i<n;i++)
        {
            cin>>x;
            for(int i=0;i<x;i++)
                cin>>a[i];
            for(int j=1;j<x;j++)
                Union(a[j-1],a[j]);
        }
        sum=0;
        for(int i=0;i<m;i++)
            if(find(parent[0])==find(parent[i]))
                sum++;//找到一个相同的根节点说明这整条边都和0有联系,都感染了.
        cout<<sum<<endl;
    }
}

意见反馈 || 任何建议

联系我(新浪)
邮箱：qianlizhihao@gmail.com