精准核酸检测

精准核酸检测

一、题目

题目描述：
为了达到新冠疫情精准防控的需要，为了避免全员核酸检测带来的浪费，需要精准圈定可能被感染的人群。
现在根据传染病流调以及大数据分析，得到了每个人之间在时间、空间上是否存在轨迹的交叉。
现在给定一组确诊人员编号(X1,X2,X3…Xn) 在所有人当中，找出哪些人需要进行核酸检测，输出需要进行核酸检测的人数。（注意:确诊病例自身不需要再做核酸检测)
需要进行核酸检测的人，是病毒传播链条上的所有人员，即有可能通过确诊病例所能传播到的所有人。
例如:A是确诊病例，A和B有接触、B和C有接触 C和D有接触，D和E有接触。那么B、C、D、E都是需要进行核酸检测的
输入描述：
第一行为总人数N
第二行为确诊病例人员编号 (确证病例人员数量 < N) ，用逗号隔开
接下来N行，每一行有N个数字，用逗号隔开，其中第i行的第个j数字表名编号i是否与编号j接触过。0表示没有接触，1表示有接触
输出描述：
输出需要做核酸检测的人数
补充说明：
人员编号从0开始
0 < N < 100

二、用并查集

import java.util.Arrays;
import java.util.Scanner;
import java.util.HashSet;

public class Main {
    private static int[] parent;

    public static void main(String[] args) {
        Scanner scanner = new Scanner(System.in);
        int N = scanner.nextInt();
        HashSet<Integer> diagnosed = new HashSet<>();
        parent = new int[N];
        scanner.nextLine();

        for (int i = 0; i < N; i++) {
            parent[i]=i;
        }

        // 读入确诊病例人员编号
        int[] diagnosedArray = Arrays.stream(scanner.nextLine().split(",")).mapToInt(Integer::parseInt).toArray();

        // 读入接触关系
        for (int i = 0; i < N; i++) {
            int[] contacts = Arrays.stream(scanner.nextLine().split(",")).mapToInt(Integer::parseInt).toArray();
            for (int j = 0; j < N; j++) {
                if (contacts[j]==1) {
                    union(i,j);
                }
            }
        }

        // 输出需要进行核酸检测的人数
        int cnt=0;
        for (int i = 0; i < parent.length; i++) {
            if(i!=parent[i]){
                cnt++;
            }
        }
        System.out.println(cnt);

    }

    private static void union(int x, int y) {
        int rootX = find(x);
        int rootY = find(y);
        if (rootX != rootY) {
            parent[rootY] = rootX;
        }
    }

    private static int find(int x) {
        if (parent[x] != x) {
            parent[x] = find(parent[x]);
        }
        return parent[x];
    }


}