并查集–The Suspects
Description
严重急性呼吸系统综合症(SARS)是一种病因不明的非典型肺炎,在2003年3月中旬被确认为全球威胁。为了尽量减少传染给他人,最好的策略是把嫌疑人和其他人分开。
在不扩散的你的疾病大学(NSYSU),有许多学生团体.同一组中的学生经常相互交流,一个学生可以加入几个小组。为了防止SARS的可能传播,NSYSU收集所有学生组的成员名单,并在其标准操作过程(SOP)中制定以下规则。
一旦一个组中的一个成员是嫌疑犯,这个组中的所有成员都是嫌疑犯。
然而,他们发现,当学生被认定为嫌疑人时,很难识别出所有的嫌疑人。你的工作是写一个程序找出所有嫌疑人。
Input
输入文件包含几种情况。每个测试用例从一行中的两个整数n和m开始,其中n是学生的数目,m是组的数目。您可以假设0<n<=30000和0<=m<=500。每个学生都由0到n−1之间的唯一整数编号,在所有情况下,最初都将学生0识别为嫌疑犯。这一行后面跟着组的m个成员列表,每组一行。每一行的开头都是一个整数k,表示组中的成员数。在成员数之后,有k个整数代表这个组中的学生。一行中的所有整数至少用一个空格分隔。
n=0和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
合并同一组的元素,遍历0所在的集合的人数
遍历0所在的集合人数可以这样
int x=find(0);
for(int i=0;i<n;i++)
{
if(find(i)==x)
num++;
}
总体代码如下
#include <iostream>
#include <cstdio>
#include <algorithm>
#include <math.h>
#include <string.h>
#include <string>
#include <stack>
#include <queue>
#include <map>
using namespace std;
const int inf=0x3f3f3f3f;
const int MAX=3e4+5;
int father[MAX];
int find(int x)
{
while (father[x] != x)
x = father[x];
return x;
}
int main()
{
int n,m;
while((scanf("%d%d",&n,&m)),n)
{
int num=0;
for(int i=0;i<n;i++)
{
father[i]=i;
}
while(m--)
{
int num,x,y,a,b;
cin>>num;
num--;
cin>>x;
while(num--)
{
cin>>y;
a=find(x);
b=find(y);
if(a!=b)
father[a]=b;
}
}
int x=find(0);
for(int i=0;i<n;i++)
{
if(find(i)==x)
{
num++;
}
}
cout<<num<<endl;
}
return 0;
}