🍭 大家好这里是清隆学长 ，一枚热爱算法的程序员

✨ 本系列打算持续跟新华为OD-C/D卷的三语言AC题解

💻 ACM银牌🥈| 多次AK大厂笔试 ｜ 编程一对一辅导

👏 感谢大家的订阅➕ 和 喜欢💗

📎在线评测链接

https://app5938.acapp.acwing.com.cn/contest/2/problem/OD1090

🌍 评测功能需要 ⇒ 订阅专栏 ⇐ 后私信联系清隆解锁～

🍓OJ题目截图

🍏 LYA的生日聚会

问题描述

LYA要举办一个生日聚会，邀请了 $n$ 位朋友参加。但是，由于最近流感病毒正在肆虐，LYA希望找出可能被感染的人群，以便及时采取防控措施。根据流行病学调查和大数据分析，得到了每个人之间是否有过密切接触的信息。现在已知一组确诊病例的编号 $(x_1,x_2,\dots ,x_m)$，请你帮助LYA找出哪些人需要进行病毒检测，并输出需要检测的人数。注意，确诊病例本身不需要再做检测。

需要进行病毒检测的人，是指在病毒传播链条上的所有人员，即所有可能被确诊病例直接或间接传染的人。例如，如果A是确诊病例，A和B有过接触，B和C有过接触，C和D有过接触，那么B、C、D都需要进行病毒检测。

输入格式

第一行包含一个正整数 $n$，表示总人数。

第二行包含若干个用逗号隔开的正整数，表示确诊病例的编号。

接下来 $n$ 行，每行包含 $n$ 个用逗号隔开的数字，其中第 $i$ 行的第 $j$ 个数字表示编号为 $i$ 的人是否与编号为 $j$ 的人有过密切接触。数字为1表示有过接触，为0表示没有接触。

输出格式

输出一个整数，表示需要进行病毒检测的人数。

样例输入

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

样例解释

在这个样例中，总共有5个人，编号分别为0到4。其中，编号为1和2的人是确诊病例。根据接触信息，我们可以发现：

• 编号为1的人和编号为0的人有过接触；
• 编号为0的人和编号为3的人有过接触；
• 编号为2的人和编号为4的人有过接触。

因此，编号为0、3、4的人都可能被感染，需要进行病毒检测。所以输出3，表示总共有3个人需要检测。

数据范围

• $0<n<100$
• 人员编号从0开始

题解

我们可以使用深度优先搜索（DFS）或广度优先搜索（BFS）来找出所有可能被感染的人。首先，将所有确诊病例加入到一个集合中，作为初始的感染者集合。然后，遍历每个人，如果这个人与任何一个已经在感染者集合中的人有过接触，就将其加入到感染者集合中。重复这个过程，直到感染者集合不再扩大为止。最后，感染者集合的大小减去初始确诊病例的数量，就是需要进行病毒检测的人数。

参考代码

• Python

n = int(input())
confirmed = set(map(int, input().split(',')))
contact = [list(map(int, input().split(','))) for _ in range(n)]

def num_to_test(n, confirmed, contact):
    infected = confirmed.copy()
    for i in range(n):
        for j in range(n):
            if contact[i][j] == 1 or contact[j][i] == 1:
                if i in infected:
                    infected.add(j)
                if j in infected:
                    infected.add(i)
    return len(infected) - len(confirmed)

print(num_to_test(n, confirmed, contact))

• Java

import java.util.*;

public class Main {
    public static void main(String[] args) {
        Scanner sc = new Scanner(System.in);
        int n = sc.nextInt();
        sc.nextLine();
        String[] s = sc.nextLine().split(",");
        Set<Integer> confirmed = new HashSet<>();
        for (String x : s) {
            confirmed.add(Integer.parseInt(x));
        }
        int[][] contact = new int[n][n];
        for (int i = 0; i < n; i++) {
            String[] line = sc.nextLine().split(",");
            for (int j = 0; j < n; j++) {
                contact[i][j] = line[j].charAt(0) - '0';
            }
        }
        System.out.println(numToTest(n, confirmed, contact));
    }

    public static int numToTest(int n, Set<Integer> confirmed, int[][] contact) {
        Set<Integer> infected = new HashSet<>(confirmed);
        for (int i = 0; i < n; i++) {
            for (int j = 0; j < n; j++) {
                if (contact[i][j] == 1 || contact[j][i] == 1) {
                    if (infected.contains(i)) {
                        infected.add(j);
                    }
                    if (infected.contains(j)) {
                        infected.add(i);
                    }
                }
            }
        }
        return infected.size() - confirmed.size();
    }
}

• Cpp

#include <iostream>
#include <vector>
#include <unordered_set>
#include <sstream>

using namespace std;

int numToTest(int n, unordered_set<int>& confirmed, vector<vector<int>>& contact) {
    unordered_set<int> infected(confirmed);
    for (int i = 0; i < n; i++) {
        for (int j = 0; j < n; j++) {
            if (contact[i][j] == 1 || contact[j][i] == 1) {
                if (infected.count(i)) {
                    infected.insert(j);
                }
                if (infected.count(j)) {
                    infected.insert(i);
                }
            }
        }
    }
    return infected.size() - confirmed.size();
}

int main() {
    int n;
    cin >> n;
    cin.ignore();
    string s;
    getline(cin, s);
    unordered_set<int> confirmed;
    stringstream ss(s);
    string x;
    while (getline(ss, x, ',')) {
        confirmed.insert(stoi(x));
    }
    vector<vector<int>> contact(n, vector<int>(n));
    for (int i = 0; i < n; i++) {
        getline(cin, s);
        stringstream ss(s);
        for (int j = 0; j < n; j++) {
            getline(ss, x, ',');
            contact[i][j] = stoi(x);
        }
    }
    cout << numToTest(n, confirmed, contact) << endl;
    return 0;
}