POJ 1611 The Suspects 【并查集基础】

POJ 1611 The Suspects

原题链接:judge传送门

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

具体思路:
并查集模板题

具体代码:

#include
#include
#include
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;
}

你可能感兴趣的:(并查集,OJ题解)