JAVA程序设计：尽量减少恶意软件的传播（LeetCode：924）

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

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

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

我们可以从初始列表中删除一个节点。如果移除这一节点将最小化 M(initial)， 则返回该节点。如果有多个节点满足条件，就返回索引最小的节点。

请注意，如果某个节点已从受感染节点的列表 initial 中删除，它以后可能仍然因恶意软件传播而受到感染。

 

示例 1：

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

输入：graph = [[1,0,0],[0,1,0],[0,0,1]], initial = [0,2]
输出：0
示例 3：

输入：graph = [[1,1,1],[1,1,1],[1,1,1]], initial = [1,2]
输出：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

思路：暴力枚举删除哪一个节点，之后通过dfs遍历并标记已经过（感染）的节点即可。复杂度O(n^2)

class Solution {
	
	private boolean[] flag;
	private List<List<Integer>> list;
	
    public int minMalwareSpread(int[][] graph, int[] initial) {
        
    	int n=graph.length;
    	flag=new boolean[n];
    	list=new ArrayList<>();
    	int ans=initial[0],num=n;
    	
    	for(int i=0;i<n;i++) {
    		list.add(new ArrayList<>());
    		for(int j=0;j<n;j++)
    			if(graph[i][j]==1)
    				list.get(i).add(j);
    	}
    	
    	for(int i=0;i<initial.length;i++) {
    		int sum=0;
    		for(int j=0;j<n;j++) flag[j]=false;
    		for(int j=0;j<initial.length;j++) {
    			if(i==j || flag[initial[j]]) continue;
    			dfs(initial[j]);
    		}
    		for(int j=0;j<n;j++)
    			if(flag[j])
    				sum++;
    		if(sum<num || sum==num && initial[i]<ans) {
    			num=sum;
    			ans=initial[i];
    		}
    	}
    	
    	return ans;
    }
    
    private void dfs(int x) {
    	
    	flag[x]=true;
    	for(int i=0;i<list.get(x).size();i++) {
    		int y=list.get(x).get(i);
    		if(flag[y]) continue;
    		dfs(y);
    	}
    }
}