Week6:戴好口罩!——并查集
题目描述
新型冠状病毒肺炎(Corona Virus Disease 2019,COVID-19),简称“新冠肺炎”,是指2019新型冠状病毒感染导致的肺炎。
如果一个感染者走入一个群体,那么这个群体需要被隔离!
小A同学被确诊为新冠感染,并且没有戴口罩!!!!!!
危!!!
时间紧迫!!!!
需要尽快找到所有和小A同学直接或者间接接触过的同学,将他们隔离,防止更大范围的扩散。
众所周知,学生的交际可能是分小团体的,一位学生可能同时参与多个小团体内。
请你编写程序解决!戴口罩!!
输入格式
多组数据,对于每组测试数据:
第一行为两个整数n和m(n = m = 0表示输入结束,不需要处理),n是学生的数量,m是学生群体的数量。0 < n <= 3e4 , 0 <= m <= 5e2
学生编号为0~n-1
小A编号为0
随后,m行,每行有一个整数num即小团体人员数量。随后有num个整数代表这个小团体的学生。
输出格式
输出要隔离的人数,每组数据的答案输出占一行
输入案例
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
输出案例
4
1
1
思路分析
实际上现在根本就没有开学,所以这道题并不存在(逃
其实看一遍题目就可以很快的发现这就是个并查集。
个人比较喜欢把并查集戏称为“找爹集”,因为对于整个流程就是找每个点的“Leader”,所有拥有同一个“Leader”的就是一个团体的。
那么思路很清晰了,就是输入数据,找爹嘛。
为了避免超时,记得使用路径优化,找起爹来更加迅捷(bushi)
#include <iostream>
#include <vector>
#include <cstring>using namespace std;int father[30002];
int rnk[30002];int findfather(int x)//找爹
{if (x == father[x]) return x;else return father[x] = findfather(father[x]);//优化路径
}void Union(int a, int b)//认爹
{a = findfather(a);b = findfather(b);if (a == b) return;/*if (rnk[a] > rnk[b]) father[b] = a, rnk[a] = (rnk[b] += rnk[a]);else father[a] = b, rnk[b] = (rnk[a] += rnk[b]);*///上面是用来维护生成树的秩的father[b] = a;
}int main()
{int n, m;while (cin >> n >> m){if (n == 0 && m == 0) break;for (int i = 0; i < n; i++){father[i] = i;rnk[i] = 1;}int num, j1, jx;for (int i = 0; i < m; i++){cin >> num;for (int j = 0; j < num; j++){if (j == 0){cin >> j1;continue;}else{cin >> jx;Union(j1, jx);}}}for (int i = 0; i < n; i++){father[i] = findfather(i);}int t = father[0];int ans = 0;for (int i = 0; i < n; i++){if (father[i] == t) ans++;}cout << ans << endl;}
}
发布评论