WEEK 6 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

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

Sample Input

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

Sample Output

4
1
1

思路:

此题采用并查集的思想,根据学生是否有直接或间接接触,对学生进行合并处理,同一团体的学生是同一集合,若两个团体学生有交集,则将两个集合合并,同时记录各个集合中的学生数目,最后输出0号学生所在集合的学生数目即可。

代码:

#include <iostream>

using namespace std;

const int maxn = 3e4 + 5;
int par[maxn], rnk[maxn];

void init(int n)
{
	for (int i = 0; i < n; i++)
		par[i] = i, rnk[i] = 1;
}
int find(int x)
{
	if (par[x] == x)
		return x;
	else
		return par[x] = find(par[x]);
}
bool unite(int x, int y)
{
	x = find(x), y = find(y);
	if (x == y)
		return false;
	par[x] = y;
	rnk[y] += rnk[x];
	return true;
}


int main()
{
	int m, n;
	while (scanf("%d%d",&n,&m)&&!(n==0&&m==0))
	{
		init(n);
		while (m--)
		{
			int num, last = -1;
			scanf("%d", &num);
			for (int i = 0; i < num; i++)
			{
				int p;
				scanf("%d", &p);
				if (last != -1)
					unite(p, last);
				last = p;
			}
		}
		printf("%d\n", rnk[find(0)]);
	}
    return 0; 
}
发布了32 篇原创文章 · 获赞 0 · 访问量 680

猜你喜欢

转载自blog.csdn.net/qq_43814559/article/details/105253696