ACM: The Suspects-并查集-解题报告

 The Suspects
Time Limit:1000MS     Memory Limit:20000KB     64bit IO Format:%lld & %llu


Description
严重急性呼吸系统综合症( SARS), 一种原因不明的非典型性肺炎,从2003年3月中旬开始被认为是全球威胁。为了减少传播给别人的机会, 最好的策略是隔离可能的患者。
在Not-Spreading-Your-Sickness大学( NSYSU), 有许多学生团体。同一组的学生经常彼此相通,一个学生可以同时加入几个小组。为了防止非典的传播,NSYSU收集了所有学生团体的成员名单。他们的标准操作程序(SOP)如下:
一旦一组中有一个可能的患者, 组内的所有成员就都是可能的患者。
然而,他们发现当一个学生被确认为可能的患者后不容易识别所有可能的患者。你的工作是编写一个程序, 发现所有可能的患者。
 
Input
输入文件包含多组数据。
对于每组测试数据:
第一行为两个整数n和m, 其中n是学生的数量, m是团体的数量。0 < n <= 300000 <= 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

简单的并查集
以每一个圈子的第一个人为根节点,把所有的人的父节点标记为根节点,然后记录该根节点的人数;
当两个节点都有根节点,并且根节点不相同的时候,合并两个圈子,根节点更改为其中一个,更新此时该根节点的人数。

AC代码:
#include"iostream"
#include"cstdio"
#include"algorithm"
#include"cmath"
#include"cstring"
using namespace std;

#define MX 1000000
int pe[MX];
int num[MX];

int find(int x) {
	return pe[x] == x ? x : (pe[x] = find(pe[x]));   
}

int main() {
	int n,m,x;
	while(~scanf("%d%d",&n,&m)) {
		if(!n&&!m) break;
		for(int i=0; i<n; i++) {
			pe[i]=i;
			num[i]=1;		//标记每个人为根节点的子节点个数 
		}
		for(int qq=1; qq<=m; qq++) {
			scanf("%d",&x);
			int st[x];
			memset(st,0,sizeof st);
			for(int i=0; i<x; i++) {
				scanf("%d",&st[i]);
				if(i>=1) {
					int root1,root2;
					root1=find(st[i-1]);
					root2=find(st[i]);
					if(root1 != root2) {//如果两个人的根节点不同 
						pe[root2] = root1;//把第二个圈子的人的根节点换为第一个圈子的根节点
						num[root1]+=num[root2];//把有同一个根节点的人数统计出来加在根节点上 
					}
				}
			}
		}
			printf("%d
",num[find(0)]);//直接搜索0的根节点的人数 
	}
}

  

 
原文地址:https://www.cnblogs.com/HDMaxfun/p/5700397.html