尽量减少恶意软件的传播

/*
 * 思路：
 * 1. 使用流的API来计算每个节点受到感染的情况。
 * 2. 对于每个初始感染节点，将其移除并计算整个网络的感染节点数。
 * 3. 选择移除后感染节点数最少的节点作为结果，若有多个，选择索引最小的节点。
 */
import java.util.*;
import java.util.stream.*;

public class MinimizeMalwareSpreadStream {
    public int minMalwareSpread(int[][] graph, int[] initial) {
        int n = graph.length;
        Map<Integer, Integer> infectedBy = IntStream.range(0, n).boxed().collect(Collectors.toMap(i -> i, i -> -1));

        Arrays.stream(initial).forEach(node -> {
            if (infectedBy.get(node) == -1) {
                boolean[] visited = new boolean[n];
                dfs(graph, node, node, visited, infectedBy);
            }
        });

        int[] count = new int[n];
        infectedBy.values().stream().filter(v -> v != -1).forEach(v -> count[v]++);

        return Arrays.stream(initial)
                .boxed()
                .min(Comparator.comparingInt((Integer node) -> count[node]).thenComparingInt(node -> node))
                .orElse(-1);
    }

    private void dfs(int[][] graph, int start, int node, boolean[] visited, Map<Integer, Integer> infectedBy) {
        if (visited[node]) return;
        visited[node] = true;
        infectedBy.put(node, start);

        IntStream.range(0, graph.length)
                .filter(i -> graph[node][i] == 1 && !visited[i])
                .forEach(i -> dfs(graph, start, i, visited, infectedBy));
    }
}