POJ - 1611 The Suspects（并查集）

还是放假第三天写的题。

题目描述

https://vjudge.net/problem/POJ-1611#author=20175101008
2003年4月16日，世界卫生组织根据包括中国内地和香港地区，加拿大、美国在内的11个国家和地区的13个实验室通力合作研究的结果，宣布重症急性呼吸综合征的病因是一种新型的冠状病毒，称为SARS冠状病毒。
很不幸，小明同学昨天晚上被确诊以经感染非典病毒，为此，校医院需要对他隔离治疗并隔离观察与他直接或间接接触者。
可以认为，同一个社团内如果有人感染病毒或可能感染病毒，那么这个社团内所有人都被认为是可能已经感染了病毒，由于时间紧迫，需要尽快找到所有可能携带病毒的同学并隔离，以防止更大范围的病毒扩散，院长请你帮忙编写程序计算需要隔离多少人
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

思路分析

这道题一看就知道是并查集，不需要解释。但是这里有一个比较有趣的操作：题目要求的是和0元素相连的点的个数，这和我们以前写的数根节点个数不同。这里采取的方法是累加：每一次合并集合都把合并双方的集合元素数加起来，放到另一个数组，对应根节点的位置，这样，大大减小了等会计算元素数时的复杂度，这是我们所要学习的。

完整代码

#include <iostream>
#include <cstring> 
const int maxn=100;
using namespace std;
int n,m;
int height[maxn];
int s[30050];
int num[30050]={0};
int find_set(int x)
{
	int r=x;
	while(r!=s[r]) r=s[r];
	int i=x,t;
	while(x!=i){
		t=s[i];
		s[i]=r;
		i=t;
	}
	return r;
}
void union_set(int x,int y)
{
	x=find_set(x);
	y=find_set(y);
	if(x!=y)
	{
		s[y]=x;
		num[x]+=num[y];
	}
}


int main()
{
	
	while(cin>>n>>m&&n+m!=0)
	{
		
		for(int i=0;i<=n;i++) s[i]=i;
		//for(int i=0;i<100;i++) s[i]=i;
		memset(height,0,sizeof(height));
		for(int i=0;i<=n;i++) num[i]=1;
		while(m--)
		{
			int t;
			cin>>t;
			int a,b;
			cin>>a;
			t--;
			while(t--)
			{
				cin>>b;
				union_set(a,b);
			} 
		}
		cout<<num[find_set(0)]<<endl;
	}
	
	return 0;
}