LeetCode.924 尽量减少恶意软件的传播 Minimize Malware Spread(并查集)

在节点网络中，只有当 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

题目很简单，很明显的并查集应用。
把各个边的两个节点放到并查集当中，处理完之后，对每个initial当中的节点进行计数，如果当前节点的连通的节点中只有他一个病变节点，统计一下这个集合节点数量。然后保留最大的那个即可。

class Solution{
private:
	int father[305];
	int node_counter[305];
	int source_counter[305];
	int find(int p)
	{
		if(p==father[p])
			return p;
		else
		{
			father[p]=find(father[p]);
			return father[p];
		}
	}
	void make_union(int p,int q)
	{
		int x,y;
		x=find(p);
		y=find(q);
		if(x!=y)
			father[x]=y;
	}
public:
    int minMalwareSpread(vector<vector<int>>& graph,vector<int>& initial)
	{
		int minus_node=0,n=graph.size();
		sort(initial.begin(),initial.end());
		int result=initial[0];
		for(int i=0;i<n;++i)
		{
			father[i]=i;
			node_counter[i]=0;
			source_counter[i]=0;
		}
		for(int i=0;i<n;++i)
		{
			for(int j=0;j<i;++j)
			{
				if(graph[i][j])
					make_union(i,j);
			}
		}
		for(int i=0;i<n;++i)
			node_counter[find(father[i])]+=1;
		for(int i=0;i<initial.size();++i)
			source_counter[find(father[initial[i]])]+=1;
		for(int i=0;i<initial.size();++i)
		{
			if(source_counter[find(father[initial[i]])]==1 && node_counter[find(father[initial[i]])]>minus_node)
			{
				minus_node=node_counter[find(father[initial[i]])];
				result=initial[i];
			}
		}
		return result;
    }
};