程序设计Week6——B-戴好口罩！

最新推荐文章于 2022-07-12 00:00:00 发布

李洋233

最新推荐文章于 2022-07-12 00:00:00 发布

阅读量258

点赞数

本文链接：https://blog.csdn.net/qq_43371596/article/details/105284951

版权

B-戴好口罩

题目描述

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

Input

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

Output

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

解题思路

本题输入的数据，其实也构成了一个图，但区别在于这幅图，并不一定是全连通的，而我们要找到所有和给定点0连通的点，而连通关系则是输入数据直接给我们的，因此我们无需再构建图的连通关系，直接用并查集合并就行，同时在合并时记录每一个集合的元素个数，合并完所有的节点后，输出0所在集合的元素个数即可。

实现代码

#include<iostream>
using namespace std;

int f[30005],r[30005];

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

void combine(int x,int y)
{
	x=find(x);
	y=find(y);
	if(x!=y)
	{
		f[x]=y;
		r[y]+=r[x];
	}
}

int main()
{
	int n,m;
	while(cin>>n>>m)
	{
		if(n==0&&m==0)
			break;
		for(int i=0;i<n;++i)
		{//初始化 
			f[i]=i;
			r[i]=1;
		}
		for(int j=0;j<m;++j)
		{
			int num,x;
			cin>>num>>x;
			for(int i=1;i<num;++i)
			{
				int a;
				cin>>a;
				combine(a,x);
			}
		}
		cout<<r[find(0)]<<endl;
	}
	return 0;
}