华为OD笔试机试 - 精准核酸检测 (c++ 2024年C卷D卷真题算法)

华为OD机试（C卷+D卷）2024真题目录(Java & c++ & python)

题目描述

为了达到新冠疫情精准防控的需要，为了避免全员核酸检测带来的浪费，需要精准圈定可能被感染的人群。

现在根据传染病流调以及大数据分析，得到了每个人之间在时间、空间上是否存在轨迹的交叉。

现在给定一组确诊人员编号 (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

输出描述

输出需要做核酸检测的人数

用例

输入

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人要进行核酸检测。

解题思路

并查集。
将有接触的人进行合并操作，纳入到同一个连通分量中。比如matrix[i]][j] == 1，即 i 和 j 就处于同一个连通分量中，需要进行合并。

另外，本题的接触关系矩阵matrix是沿对角线对称的，因此只需要遍历对角线一边即可。

当遍历完所有接触关系后，就可以求解每一个连通分量中的节点数，即每个接触群体的人数，求解原理如下：

并查集底层的fa数组，fa数组索引代表每个节点，fa数组元素代表对应索引的节点的根节点，而同一个连通分量中的节点的根都是相同的，因此，我们需要对fa每一个数组索引找一下根，这里可以使用并查集的find操作（递归实现），最后统计同一个根下的节点数量，即为同一个接触群体的人数。

当每个接触群体人数求解出来后，我们只需要统计”确诊病例人员编号“对应的根（连通分量）下的人数即可。

最后的统计的总人数需要减去确诊病例的数量，因为题目说：

确诊病例自身不需要再做核酸检测

本题需要注意的是，有可能多个确诊病人在同一个连通分量重，此时需要注意避免重复统计。

参考代码

#include <bits/stdc++.h>

using namespace std;

// 并查集实现
class UnionFindSet {
public:
    vector<int> parent;

    explicit UnionFindSet(int n) {
        parent.resize(n);
        iota(parent.begin(), parent.end(), 0);
    };

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

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

// 按照指定分隔符separator分割字符串s，并且分隔后的各部分token转为整型数值，录入一个vector中
vector<int> split(const string &s, char separator) {
    stringstream ss(s);
    string token;
    vector<int> res;
    while (getline(ss, token, separator)) {
        res.push_back(stoi(token));
    }
    return res;
}

int main() {
    // 输入获取
    int n;
    cin >> n;

    string s;
    cin >> s;
    vector<int> confirmed = split(s, ',');

    vector<vector<int>> matrix(n, vector<int>(n));
    for (int i = 0; i < n; i++) {
        string tmp;
        cin >> tmp;
        matrix[i] = split(tmp, ',');
    }

    // 算法逻辑
    UnionFindSet ufs(n);

    for (int i = 0; i < n; i++) {
        for (int j = i; j < n; j++) {
            if (matrix[i][j] == 1) {
                // 有过接触的人进行合并
                ufs.merge(i, j);
            }
        }
    }

    // 统计每个接触群体（连通分量）中的人数
    vector<int> groupCount(n, 0);
    for (int i = 0; i < n; i++) {
        int root = ufs.find(i);
        groupCount[root]++;
    }

    // 记录已统计过的感染群体
    unordered_set<int> confirmedGroups;

    // 将有感染者的接触群体的人数统计出来
    int ans = 0;
    for (int person : confirmed) {
        int root = ufs.find(person);
        if (confirmedGroups.count(root) == 0) {
            confirmedGroups.insert(root);
            ans += groupCount[root];
        }
    }

    // 最终需要做核酸的人数，不包括已感染的人
    cout << (ans - confirmed.size()) << endl;

    return 0;
}