POJ1611 The Suspects(并查集)

第一次学并查集，总算对它有了一定的了解，基本就是连通的集合的问题

参考博客：基本概念   结合一些具体问题

题意：

学校中一共有m个学生，n个组，学生编号从0~m-1，现在学生0感冒了，如果组中有一人感冒，剩下的人都会感冒，给出每个组的具体学生序号，求出有多少人会感冒

要点：

并查集裸题，要点是先将所有组合并，最后遍历找于0同一根节点的数量即为感冒人数，要注意0不一定是最终的根节点

15310408

Seasonal

1611

Accepted

404K

16MS

C++

915B

2016-03-24 22:21:30

#include<stdio.h>
#include<string.h>
#include<stdlib.h>
#define maxn 40000
int p[maxn], rank[maxn];
int m, n;

void init()
{
	for (int i = 0; i < m; i++)
	{
		p[i] = i;
		rank[i] = 0;//记录树的深度
	}
}
int find(int x)//找到根节点
{
	if (x == p[x]) return x;
	return p[x] = find(p[x]);//状态压缩，使所有节点都指向根节点
}
void merge(int x,int y)//合并集合
{
	x = find(x);
	y = find(y);
	if (x == y) return;
	if (rank[x] > rank[y])//将深度小的树指向深度大的树
		p[y] = x;
	else
	{
		p[x] = y;
		if (rank[x] == rank[y])//深度如果相同，深度大的树深度要+1
			rank[y]++;
	}
}

int main()
{
	while (~scanf("%d%d", &m, &n), n + m)
	{
		init();
		while (n--)
		{
			int t,x,y;
			scanf("%d", &t);
			scanf("%d", &x);//每组的第一个数
			for (int i = 1; i < t; i++)
			{
				scanf("%d", &y);
				merge(x, y);//将每组的后续数都与第一个数合并
			}
		}
		int sum = 0;
		int xx = find(0);//注意0不一定是根节点，可能只是在树中
		for (int i = 0; i < m; i++)
			if (find(i) == xx)
				sum++;
		printf("%d\n", sum);
	}
	return 0;
}