Week6:戴好口罩!——并查集

题目描述
新型冠状病毒肺炎(Corona Virus Disease 2019,COVID-19),简称“新冠肺炎”,是指2019新型冠状病毒感染导致的肺炎。
如果一个感染者走入一个群体,那么这个群体需要被隔离!
小A同学被确诊为新冠感染,并且没有戴口罩!!!!!!
危!!!
时间紧迫!!!!
需要尽快找到所有和小A同学直接或者间接接触过的同学,将他们隔离,防止更大范围的扩散。
众所周知,学生的交际可能是分小团体的,一位学生可能同时参与多个小团体内。
请你编写程序解决!戴口罩!!

输入格式
多组数据,对于每组测试数据:
第一行为两个整数n和m(n = m = 0表示输入结束,不需要处理),n是学生的数量,m是学生群体的数量。0 < n <= 3e4 , 0 <= m <= 5e2
学生编号为0~n-1
小A编号为0
随后,m行,每行有一个整数num即小团体人员数量。随后有num个整数代表这个小团体的学生。

输出格式
输出要隔离的人数,每组数据的答案输出占一行

输入案例

100 4
2 1 2
5 10 13 11 12 14
2 0 1
2 99 2
200 2
1 5
5 1 2 3 4 5
1 0
0 0

输出案例

4
1
1

思路分析
实际上现在根本就没有开学,所以这道题并不存在(逃

其实看一遍题目就可以很快的发现这就是个并查集。
个人比较喜欢把并查集戏称为“找爹集”,因为对于整个流程就是找每个点的“Leader”,所有拥有同一个“Leader”的就是一个团体的。

那么思路很清晰了,就是输入数据,找爹嘛。
为了避免超时,记得使用路径优化,找起爹来更加迅捷(bushi)

#include <iostream>
#include <vector>
#include <cstring>

using namespace std;

int father[30002];
int rnk[30002];

int findfather(int x)//找爹
{
	if (x == father[x]) return x;
	else return father[x] = findfather(father[x]);//优化路径
}

void Union(int a, int b)//认爹
{
	a = findfather(a);
	b = findfather(b);
	if (a == b) return;

	/*if (rnk[a] > rnk[b]) father[b] = a, rnk[a] = (rnk[b] += rnk[a]);
	else father[a] = b, rnk[b] = (rnk[a] += rnk[b]);*/
	//上面是用来维护生成树的秩的

	father[b] = a;
}

int main()
{
	int n, m;
	while (cin >> n >> m)
	{
		if (n == 0 && m == 0) break;

		for (int i = 0; i < n; i++)
		{
			father[i] = i;
			rnk[i] = 1;
		}

		int num, j1, jx;
		for (int i = 0; i < m; i++)
		{
			cin >> num;
			for (int j = 0; j < num; j++)
			{
				if (j == 0)
				{
					cin >> j1;
					continue;
				}
				else
				{
					cin >> jx;
					Union(j1, jx);
				}
			}
		}

		for (int i = 0; i < n; i++)
		{
			father[i] = findfather(i);
		}

		int t = father[0];
		int ans = 0;
		for (int i = 0; i < n; i++)
		{
			if (father[i] == t) ans++;
		}

		cout << ans << endl;

	}
}

发布了21 篇原创文章 · 获赞 3 · 访问量 403

猜你喜欢

转载自blog.csdn.net/qq_44506233/article/details/105282571
今日推荐