题目描述
为了达到新冠疫情精准防控的需要,为了避免全员核酸检测带来的浪费,需要精准圈定可能被感染的人群。现在根据传染病流调以及大数据分析,得到了每个人之间在时间、空间上是否存在轨迹交叉。
现在给定一组确诊人员编号(X1,X2,X3,...,Xn),在所有人当中,找出哪些人需要进行核酸检测,输出需要进行核酸检测的人数。(注意:确诊病例自身不需要再做核酸检测)
需要进行核酸检测的人,是病毒传播链条上的所有人员,即有可能通过确诊病例所能传播到的所有人。
例如:A是确诊病例,A和B有接触、B和C有接触、C和D有接触、D和E有接触,那么B\\C\\D\\E都是需要进行核酸检测的人。
输入描述
- 第一行为总人数 N
- 第二行为确认病例人员编号(确诊病例人员数量 < N),用逗号分割
- 第三行开始,为一个 N * N 的矩阵,表示每个人员之间是否有接触,0表示没有接触,1表示有接触。
输出描述
整数:需要做核酸检测的人数
备注
人员编号从0开始
0 < N < 100
用例1
输入
5
1,2
1,1,0,1,0
1,1,0,0,0
0,0,1,0,1
1,0,0,1,0
0,0,1,0,1
输出
3
说明
- 编号为1、2号的人员,为确诊病例。
- 1号和0号有接触,0号和3号有接触。
- 2号和4号有接触。
所以,需要做核酸检测的人是0号、3号、4号,总计3人需要进行核酸检测。
题目解析
本题可以用并查集解题。
将有接触的人进行合并放入同一个连通分量中,如果mat[i][j]==1,则i,j需要进行合并;
所有节点遍历完之后,可以计算确诊人员所在的根节点下的元素的个数,要注意给出的确诊
人员可能在同一个连通分量中,注意不能重复计算。
代码
c++
#include <bits/stdc++.h>
#include <algorithm>
#include <math.h>
using namespace std;
vector<int> split(string s, char sep) {
vector<int> vec;
stringstream ss(s);
string tmp;
while(getl