poj1611并查集求元素所在集合的元素总数

最新推荐文章于 2024-07-09 11:21:53 发布

yuanba_xs

最新推荐文章于 2024-07-09 11:21:53 发布

阅读量591

点赞数 1

分类专栏：并查集文章标签：并查集

本文链接：https://blog.youkuaiyun.com/yuanba_xs/article/details/55666820

版权

博客内容介绍了如何运用并查集解决poj1611问题，通过合并相同团队的成员并计算每个集合的元素数量来找出总人数。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

并查集应用：求某元素所在集合的元素个数

题目：已知有n个人，m个团队，已知0号选手有病毒，病毒会传染给他所在任一小组中的人。
求最后有病毒的人的个数。

思路1:对相同团队的人进行合并，最后输出与0在同一集合的元素个数。
即是否满足find(0)==find(i)。

思路2：对相同团队的人进行合并，合并过程中对集合元素个数num也和并，最后找到0的祖先的num即可

思路1：

#include <cstdio>
#include <cstring>
#define N 30005

int f[N];
int n, m;

void init()
{
	for (int i = 0; i < n; i++)
		f[i] = i;
}

int find(int x)
{
	return x == f[x] ? x : f[x] = find(f[x]);
}

void merge(int x, int y)
{
	int t1 = find(x), t2 = find(y);
	if (t1 != t2)
		f[t2] = t1;
}

int main()
{
	while (~scanf("%d%d", &n, &m) && (n || m))
	{
		int t, u, v;
		init();
		for (int i = 0; i < m; i++)
		{
			scanf("%d", &t);
			if (t > 0)
			{
				scanf("%d", &u);
				for (int i