尽量减少恶意软件的传播 IIJava

文章发布较早，内容可能过时，阅读注意甄别。

# 题目

给定一个由 n 个节点组成的网络，用 n x n 个邻接矩阵 graph 表示。在节点网络中，只有当 graph[i][j] = 1 时，节点 i 能够直接连接到另一个节点 j。

一些节点 initial 最初被恶意软件感染。只要两个节点直接连接，且其中至少一个节点受到恶意软件的感染，那么两个节点都将被恶意软件感染。这种恶意软件的传播将继续，直到没有更多的节点可以被这种方式感染。

假设 M(initial) 是在恶意软件停止传播之后，整个网络中感染恶意软件的最终节点数。

我们可以从 initial 中删除一个节点，并完全移除该节点以及从该节点到任何其他节点的任何连接。

请返回移除后能够使 M(initial) 最小化的节点。如果有多个节点满足条件，返回索引最小的节点。

示例 1：

输入：graph = [[1,1,0],[1,1,0],[0,0,1]], initial = [0,1]
输出：0

示例 2：

输入：graph = [[1,1,0],[1,1,1],[0,1,1]], initial = [0,1]
输出：1

示例 3：

输入：graph = [[1,1,0,0],[1,1,1,0],[0,1,1,1],[0,0,1,1]], initial = [0,1]
输出：1

提示：

n == graph.length
n == graph[i].length
2 <= n <= 300
graph[i][j] 是 0 或 1.
graph[i][j] == graph[j][i]
graph[i][i] == 1
1 <= initial.length < n
0 <= initial[i] <= n - 1
initial 中每个整数都不同

# 思路

// 暴力枚举可删除的点或者并查集合并未删除的点均可~

# 解法


class Solution {
    // 暴力枚举可删除的点或者并查集合并未删除的点均可~


    public int minMalwareSpread(int[][] graph, int[] initial) {
        int inf = 0x3f3f3f3f, ans = inf, cnt = inf;
        Arrays.sort(initial);
        for (int x : initial) {
            int cur = bfs(graph, x, initial);
            if (cur < cnt) {
                cnt = cur;
                ans = x; 
            }
        }
        return ans;
    }

    private int bfs(int[][] graph, int del, int[] initial) {
        Deque<Integer> q = new ArrayDeque<>();
        Set<Integer> set = new HashSet<>();
        int cnt = 0;
        for (int x : initial) {
            if (x != del) {
                set.add(x);
                q.add(x);
                cnt++;
            }
        }
        while (!q.isEmpty()) {
            int u = q.poll();
            for (int i = 0; i < graph[u].length; i++) {
                if (graph[u][i] == 1 && i != del) {
                    if (set.add(i)) {
                        q.add(i);
                        cnt++;
                    }
                }
            }
        }
        return cnt;
    }
}

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43

# 总结

分析出几种情况，然后分别对各个情况实现

#Java