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

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

3
一些节点 initial 最初被恶意软件感染。
4

5
只要两个节点直接连接，且其中至少一个节点受到恶意软件的感染，那么两个节点都将被恶意软件感染。
6

7
这种恶意软件的传播将继续，直到没有更多的节点可以被这种方式感染。
8

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

11
如果从初始列表中移除某一节点能够最小化 M(initial)，
12

13
返回该节点。如果有多个节点满足条件，就返回索引最小的节点。
14

15
请注意，如果某个节点已从受感染节点的列表 initial 中删除，
16

17
它以后可能仍然因恶意软件传播而受到感染。
18

19

20

21
示例 1：
22

23
输入：graph = [[1,1,0],[1,1,0],[0,0,1]], initial = [0,1]
24
输出：0
25
示例 2：
26

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

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

34

35
提示：
36

37
1 < graph.length = graph[0].length <= 300
38
0 <= graph[i][j] == graph[j][i] <= 1
39
graph[i][i] == 1
40
1 <= initial.length < graph.length
41
0 <= initial[i] < graph.length

1
class UnionFind {
2
public:
3
    vector<int> data;
4
    vector<int> cnt; // 每个节点连通块大小
5

6
    void init(int n) {
7
        data.resize(n, 0);
8
        for(int i = 0; i < n; i++) {
9
            data[i] = i;
10
        }
11
        cnt.resize(n, 1);
12
    }
13

14
    int find(int x) {
15
        /* int root = x;
16
        while (data[root] >= 0) root = data[root];
17

18
        //路径压缩
19
        while (x != root) {
20
            int tmp = data[x]; //tmp 指向 x 的父节点
21
            data[x] = root;     //挂到根节点下
22
            x = tmp;
23
        }
24

25
        return root; //返回根节点的编号。 */
26
        //上述的优化会超时！
27
        if(data[x] != x) {
28
            return find(data[x]);
29
        } else {
30
            return x;
31
        }
32
    }
33

34
    void Union(int x, int y) {
35
        int p = find(x);
36
        int q = find(y);
37

38
        data[p] = q;
39
        if (p != q) {
40
            cnt[q] += cnt[p];
41
        }
42
    }
43

44
    int GetCnt(int x) {
45
        return cnt[find(x)];
46
    }
47
};
48

49
class Solution {
50
public:
51
    int minMalwareSpread(vector<vector<int>>& graph, vector<int>& initial) {
52
        UnionFind u;
53
        int n = graph.size(), m = graph[0].size();
54
        u.init(n);
55

56
        sort(initial.begin(), initial.end());
57

58
        for(int i = 0; i < n; i++) {
59
            for (int j = i + 1; j < m; j++) {
60
                if (graph[i][j] == 1) u.Union(i, j);
61

62
            }
63
        }
64

65
        // 计算initial中共连通块元素个数
66
        map<int, int> count;
67
        for (auto c : initial) {
68
            count[u.find(c)]++;
69
        }
70

71
        int ans = -1;
72
        int res = -1;
73
        for (auto c : initial) {
74
            if (count[u.find(c)] == 1) {
75
                int size = u.GetCnt(c);
76
                if (size > res) {
77
                    res = size;
78
                    ans = c;
79
                }
80
            }
81
        }
82
        if (ans == -1) return initial[0];
83
        return ans;
84
    }
85
};