POJ 1611 / ZOJ 1789: The Suspects - 并查集

学姐给我这个菜鸟讲的题让我第一次接触到并查集的使用。

本题是并查集的基础应用题。

题目大意：

SARS(非典型肺炎)传播得非常厉害，其中最有效的办法是隔离那些患病、和患病者接触的人。现在有几个学习小组，每小组有几个学生，一个学生可能会参加多个小组。小组中只要有一个人得病，其余的都是嫌疑人。现在已知这些小组的人员，且0号学生已经患病，求一共有多少个嫌疑人。

分析：每个小组的人员属于同一个集合。在根节点记录每个集合的个数num[i]，然后找到0所属的根节点，num[findSet(0)]即为所有的嫌疑人数。

输入：

每个案例

第一行为志愿者人数n和组成小组数量m，

接下来是n行，

每行第一个数k是当前小组的成员人数，其余数为各个小组中成员的编号。

感觉这个同志的blog里写的很详细：http://cavenkaka.iteye.com/blog/1171373

#include <iostream>
#include <cstdio>
#include <cstdlib>
using namespace std;
const int Max=30001;

int p[Max],rank[Max];
int num[Max];

int Find_Set(int x)  //路径压缩:经过"递推"找到祖先节点后，"回溯"的时候顺便将它的子孙节点都直接指向祖先
{
	if(x!=p[x])
		p[x]=Find_Set(p[x]);
	return p[x];
}
void Union(int x,int y)  //按秩合并：合并的时候将元素少的集合合并到元素多的集合中，这样合并之后树的高度会相对较小
{
	x=Find_Set(x);
	y=Find_Set(y);
	if(x==y)
		return;
	if(rank[x]>rank[y])
	{
		p[y]=x;
		num[x]+=num[y];
	}
	else
	{
		p[x]=y;
		if(rank[x]=rank[y])
			rank[x]++;
		num[y]+=num[x];
	}
}
int main()
{
	int i,j,k,n,m;
	int a,b;
	int s1,s2;
	while(true)
	{
		scanf("%d %d",&n,&m);
		if(n==0&&m==0)
			break;
		for(i=0;i<n;i++)
		{
			p[i]=i;
			rank[i]=0;
			num[i]=1;
		}
		for(i=0;i<m;i++)
		{
			scanf("%d%d",&k,&a);
			for(j=1;j<k;j++)
			{
				scanf("%d",&b);
				s1=Find_Set(a);
				s2=Find_Set(b);
				Union(s1,s2);
			}
		}
		i=Find_Set(0);
		printf("%d\n",num[i]);
	}
	return 0;
}

***************************************************************** ***************************************************************** ***************************************************************** *****************************************

1. 合并两个集合

void merge(int a, int b) 
{
par[get_par(a)] = get_par(b);
}

2. 查询一个元素在哪个集合

int get_par(int a) 
{
if ( par[a] != a ) 
par[a] = get_par(par[a]);
return par[a];
}

3. 查询两个元素是否属于同一集合

int query(int a, int b) 
{
return get_par(a) == get_par(b);
}