程序设计思维与实践 Week6 作业 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在同一个并查集结构中的人数。题目的输入方式有些特殊,分组输入:对于每个组,其成员肯定在同一个并查集中,因此在输入时即可对其中的成员进行处理。最后统计与0在同一个并查集中的人数即可。

进行了路径压缩,使得每个元素的father直接为其所在的并查集的代表元,降低了时间复杂度。

#include<iostream>
using namespace std;
int n,m,sum,ans,father[50010];
int find(int x)
{
	if(father[x]!=x)
	father[x]=find(father[x]);
	return father[x];
}
void unionn(int a,int b)
{
	int fa=find(a);
	int fb=find(b);
	father[fa]=fb;
}
int main()
{
	int a,b;
	while(1)
	{
		cin>>n>>m;
		if(n==0&&m==0) break;
		ans=0;
		for(int i=0;i<n;i++)
		father[i]=i; 
		for(int i=1;i<=m;i++)
		{
			cin>>sum;
			cin>>a;a=find(a);
			for(int j=2;j<=sum;j++)
			{
				cin>>b;
				b=find(b);
				if(a!=b)
				unionn(a,b);
			}
		}
		int flag=find(0);
		for(int i=0;i<n;i++)
		if(find(i)==flag)
		ans++;
		cout<<ans<<endl;
	}
	return 0;
}
发布了277 篇原创文章 · 获赞 222 · 访问量 16万+

猜你喜欢

转载自blog.csdn.net/cax1165/article/details/105107050