53、尽量减少恶意软件的传播 II

题目描述：
(这个问题与 尽量减少恶意软件的传播 是一样的，不同之处用粗体表示。)

在节点网络中，只有当 graph[i][j] = 1 时，每个节点 i 能够直接连接到另一个节点 j。

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

假设 M(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

提示：

1 < graph.length = graph[0].length <= 300
0 <= graph[i][j] == graph[j][i] <= 1
graph[i][i] = 1
1 <= initial.length < graph.length
0 <= initial[i] < graph.length

来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/minimize-malware-spread-ii
著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。

使用笨方法，每次新建一个，对于initial中的病毒节点去掉相邻的，然后计算其他病毒节点的感染个数，找出最小的即可，一直卡在dfs中。。。

class Solution {
    int gets = 0;
    public int minMalwareSpread(int[][] graph, int[] initial) {
        // 1. Color each component.
        // colors[node] = the color of this node.
        int N = graph.length;
          int tem = N;
        Arrays.sort(initial);
//        尝试每次移除其中一个病毒，然后计算被感染的
        int ans = initial[0];
        for (int i = 0; i < initial.length; i++) {
            int i1 = initial[i];
//            重新建立一个graph
            int [][] graphnew = new int[N][N];
            for (int j = 0; j < graphnew.length; j++) {
                graphnew[j] = graph[j].clone();
            }
//            将当前的节点断开
            for (int j = 0; j < N; j++) {
                if(graphnew[j][initial[i]] == 1){
                    graphnew [initial[i]][j] = 0;
                    graphnew[j][initial[i]] = 0;
                }
            }
//            开始侵染
            int tem1 = 0;
                int colors[] = new int[N];
                Arrays.fill(colors,-1);
                gets = 0;
            for (int j = 0; j < initial.length; j++) {
                int temss  = gets;  
                for (int k = 0; k < N; k++) {
                    if(graphnew[initial[j]][k] == 1 && colors[k] == -1){   
                        // colors[k] = 0;
                        tem1 += dfs(graphnew,colors,k,0);
                    }
                } 
            }
            if(gets < tem){
                tem =  gets;
                ans = initial[i];
            }
        }
        return ans;
    }
    public int dfs(int[][] graph, int[] colors, int node, int color) {
        gets ++;
        colors[node] = 0;
        int t = 0;
        for (int nei = 0; nei < graph.length; ++nei){
            if (graph[node][nei] == 1 && colors[nei] == -1 ){
                              
                                t += dfs(graph, colors, nei, color);

            }
        }
        
        return t + 1;
    }
}