题意
新型冠状病毒肺炎(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:
输出要隔离的人数,每组数据的答案输出占一行
样例:
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
OUTPUT:
4
1
1
思路
- 分析此题,我们可以使用并查集进行求解,首先就是并查集的初始化,将所有人都变成一个集合,然后每一小组的人进行并查集的合并;
- 如果一个并查集中出现了0 ,而且该集合中的元素如果也出现在其他集合中,那么在进行并查集的合并的时候,就已经更新了,合并了;
- 最后去统计和0号元素一样的值的元素个数即可;
- 此处使用的数组描述的并查集,能够在合并时将所有的值进行更新;
总结
此题使用数组描述的并查集的时候,能够在每次进行合并的时候,将所有属于此并查集的所有元素更新为同一个值;
此处应用的数组并查集在数据范围过大的时候,合并更新的时候复杂度会变得较高,可能会超时;
代码
#include<iostream>
#include<vector>
using namespace std;
int N;
int a[30000];
void init()//并查集的初始化
{
for (int i = 0; i < N; i++)
a[i] = i;
}
int find(int x)//并查集的查询
{
return a[x];
}
void unite(int x, int y)
{
y = find(y);
x = find(x);
for (int i = 0; i < N; i++)
{
if (a[i] == y)
a[i] = x;
}
}
int main()
{
while (cin >> N)
{
if (N == 0)
break;
//并查集的初始化
init();
int m;
cin >> m;
int sum = 0;
for (int i = 0; i < m; i++)
{
int sum;
cin >> sum;
int* p = new int[sum];
for (int i = 0; i < sum; i++)
{
cin >> p[i];
}
if (sum >= 2)
{//进行并查集的合并
for (int i = 0; i < sum; i++)
if (i < sum - 1)
unite(p[i], p[i + 1]);
}
}
if (m == 0) {
cout << 1 << endl;
}
else {
//去寻找感染人数
for (int i = 0; i < N; i++)
{
if (a[0] == a[i])
sum++;
}
cout << sum << endl;
}
}
return 0;
}