题意:
如果一个感染者走入一个群体,那么这个群体需要被隔离!
小A同学被确诊为新冠感染,并且没有戴口罩!!!!!!
需要尽快找到所有和小A同学直接或者间接接触过的同学,将他们隔离,防止更大范围的扩散。
众所周知,学生的交际可能是分小团体的,一位学生可能同时参与多个小团体内。
输入要求:
多组数据。每组数据第一行为两个整数 n 和 m(n = m = 0表示输入结束,不需要处理),n是学生数量,m是学生群体数量,0 < n <= 3e4 , 0 <= m <= 5e2。
学生编号为 0 ~ n - 1;小 A 同学的编号为 0。
之后输入m行,每行有一个整数 num 表示小群体人数,随后又 num 个整数代表该群体的学生编号。
输出要求:
输出要隔离的人数,每组数据的答案输出占一行
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
思路:
- 该题是个显然的并查集应用,所有接触过的人都放在一类人中。每个接触都对并查集进行 num 次 unite 操作。
- 并查集使用一个数组 par[n] 来表示,其中查找函数 find(int x) 是一个递归函数,若 par[x] == x,则返回 par[x],反之则返回 find(par[x]),并且将 find(par[x]) 赋值给 par[x]。合并函数 unite(int x, int y) 函数,首先找到两者各自的代表数 x_ 和 y_,在将两者合并(par[y_] = x_;),由于该题需要输出隔离的人数,即输出某个集合的元素个数,故又定义了一个数组 rnk 用于记录集合的人数。
代码:
#include <iostream>
using namespace std;
int par[30005], rnk[30005];
int group[30005];//用于存放一个小团体的人
int find(int x) {
if (par[x] == x)return par[x];
return par[x] = find(par[x]);
}
void unite(int x, int y) {
int x_ = find(x);
int y_ = find(y);
if (x_ == y_)return;
if (x_ < y_) {
par[y_] = x_;
rnk[x_] = rnk[x_] + rnk[y_];
}
else {
par[x_] = y_;
rnk[y_] = rnk[y_] + rnk[x_];
}
}
int main()
{
int n, m, num;
while (1) {
scanf_s("%d%d", &n, &m);
if (m == n && m == 0)break;
for (int i = 0; i < n; i++) {
par[i] = i;
rnk[i] = 1;
}
for (int i = 0; i < m; i++) {
cin >> num;
for (int j = 0; j < num; j++) {
cin >> group[j];
if (j > 0) {
unite(group[j - 1], group[j]);
}
}
}
cout << rnk[0] << endl;
}
}