华为od机考_精准核酸检测_Java（深度优先搜索）

华为od机考_精准核酸检测_Java

题目

为了达到新冠疫情精准防控的需要，为了避免全员核酸检测带来的浪费，需要精准還定可能被感染的人群。 现在根据传染病流调以及大数据分析，得到了每个人之间在时间、空间上是否存在轨迹的交叉。 现在给定一组确诊人员编号（X1,X2,X3，…Xn），在所有人当中，找出哪些人需要进行核酸检测，输出需要进行核酸检测的人数。（注意：确诊病例自身不需要再做核酸检测） 需要进行核酸检测的人，是病毒传擂链条上的所有人员，即有可能通过确诊病例所能传强到的所有人。

例如：A是确诊病例，A和B有接触、B和C有接館、C和D有接触、D和E有接鯰，那么BICIDE都是需要进行核酸检测的人。

输入描述： 第一行为总人效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人需要进行核酸检测。

代码

public class Test6 {
    //图的深度优先搜索dfs
    public static void main(String[] args) {
        Scanner input = new Scanner(System.in);
        int n = input.nextInt();//总人数
        input.nextLine();
        String[] confirmedCasesStrings = input.nextLine().split(","); //确诊病例
        Set<Integer> confirmedCasesSet = new HashSet<Integer>();//确诊病例set
        for(String c : confirmedCasesStrings){
            confirmedCasesSet.add(Integer.parseInt(c));
        }
        int[][] matrix = new int[n][n]; //存放人员关联矩阵
        for(int i =0; i<n; i++){
            String[] s1 = input.nextLine().split(",");
            for(int j = 0; j < n; j++){
                matrix[i][j] = Integer.parseInt(s1[j]);
            }
        }

        boolean[] visied = new boolean[n];//记录已遍历人员，防止重复遍历
        Set<Integer> result = new HashSet<>();//存放需要做核酸的人员
        // 深度优先遍历每一个确诊人员
        for(Integer g: confirmedCasesSet){
            dfs(matrix, g, visied, confirmedCasesSet, result);
        }
        System.out.println(result.size());
    }
    // 深度优先遍历
    public static void dfs(int[][] matrix, Integer person, boolean[] visied, Set<Integer> confirmedCasesSet, Set<Integer> result){
        visied[person] = true;
        for(int i = 0; i < matrix.length; i++) {
            if(matrix[person][i] == 1 && !visied[i] && !confirmedCasesSet.contains(i)){
                result.add(i);
                dfs(matrix, i, visied, confirmedCasesSet, result);
            }
        }
    }
}