LC每日一题｜20240416 - 924. 尽量减少恶意软件的传播

LC每日一题｜20240416 - 924. 尽量减少恶意软件的传播

给出了一个由 n 个节点组成的网络，用 n × n 个邻接矩阵图 graph 表示。在节点网络中，当 graph[i][j] = 1 时，表示节点 i 能够直接连接到另一个节点 j。 

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

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

如果从 initial 中移除某一节点能够最小化 M(initial)， 返回该节点。如果有多个节点满足条件，就返回索引最小的节点。

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

提示：

• n == graph.length
• n == graph[i].length
• 2 <= n <= 300
• graph[i][j] == 0 或 1.
• graph[i][j] == graph[j][i]
• graph[i][i] == 1
• 1 <= initial.length <= n
• 0 <= initial[i] <= n - 1
• initial 中所有整数均不重复

题目等级：Hard

解题思路

BFS暴力，其他的没什么可说的。

这题目Hard真的是拜题干所赐，可以说是leetcode上少有的垃圾题...

AC代码

class Solution {
    fun minMalwareSpread(graph: Array<IntArray>, initial: IntArray): Int {
        val map = Array<ArrayList<Int>>(graph.size) { arrayListOf() }
        for (i in graph.indices) for (j in graph[i].indices) {
            if (i != j && graph[i][j] == 1) map[i].add(j)
        }

        initial.sort()
        val c = IntArray(initial.size)
        for (i in initial.indices) {
            val visited = BooleanArray(graph.size)
            val queue = ArrayDeque<Int>()
            var count = 0
            for (j in initial.indices) if (i != j) {
                queue.addLast(initial[j])
                visited[initial[j]] = true
                count++
            }

            while (queue.isNotEmpty()) {
                repeat(queue.size) {
                    val n = queue.removeFirst()
                    map[n].forEach {
                        if (!visited[it]) {
                            visited[it] = true
                            count ++
                            queue.addLast(it)
                        }
                    }
                }
            }

            c[i] = count
        }
    
        return initial[res.indexOf(c.min()!!)]
    }
}