924. 尽量减少恶意软件的传播

题目描述：

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

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

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

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

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

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

示例：

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

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

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

 分析：

对于此类题目，类似于找出节点之间的关系，都可以考虑用并查集。

那么首先肯定就是构建一个并查集，感染就是把一整条连通分量全部感染，那么就需要找到所在连通分量的长度，

如何找呢？肯定得在构建的时候就保存起来，因为不保存的话，后面去找只能往上，不能往下，所以构建的时候就保存起来

最后就是对 initial 中每个元素找祖宗，长度比较，保存节点

上代码：

class Solution:
    def minMalwareSpread(self, graph: List[List[int]], initial: List[int]) -> int:
        pre = list(range(len(graph)))
        sz = [1] * len(graph)
        def find(x):
            while x != pre[x]:
                x = pre[x]
            return x
        
        def join(x,y):
            fx = find(x)
            fy = find(y)
            pre[fx] = fy
            sz[fy] += sz[fx]

        def size(x):
            return sz[find(x)]

        for i in range(len(graph)):
            for j in range(i):
                if graph[i][j] == 1:
                     join(i, j)
        
        count = collections.defaultdict(int)
        for i in initial:
            count[find(i)] += 1
        initial.sort()
        res = (-1, initial[0])
        for i in range(len(initial)):
            root = find(initial[i])
            if count[root] == 1:
                if size(root) > res[0]:
                    res = (size(root), initial[i])
        return res[1]