POJ 1611 The Suspects 【并查集基础】

POJ 1611 The Suspects

原题链接：judge传送门

题目大意：
有n个人，m个社团，有一种病毒具有传染性，假设一个人可能感染，那么它所属社团的所有人都可能感染，
假设0号这个人疑似感染了病毒，问总共有多少人可能感染病毒

具体思路：
并查集模板题

具体代码：

#include<iostream>
#include<cstdio>
#include<algorithm>
using namespace std;
int const N = 3e4+10;
int fa[N];
int n, m;

void init()
{
	for (int i = 0; i <= n; i++)
		fa[i] = i;
}
int find(int i)
{
	if (i != fa[i])
		fa[i] = find(fa[i]);
	return fa[i];
}
void unite(int i, int j)
{
	int x = find(i);
	int y = find(j);
	if (x == y)return;
	fa[x] = y;
}

int main()
{
	while (scanf("%d%d", &n, &m)&& (n || m))
	{
		init();
		for (int i = 1; i <= m; i++)
		{
			int num;
			scanf("%d", &num);
			int x, y;
			for (int j = 1; j <= num; j++)
			{
				scanf("%d", &y);
				if (j == 1) {
					x = y;
					continue;
				}
				unite(x, y);
			}
		}
		
		int sum=0;
		int fi = find(0);
		for (int i = 0; i < n; i++)
			if (fi == find(i))sum++;
		printf("%d\n", sum);
	}
	return 0;
}