初学python记录:力扣928. 尽量减少恶意软件的传播 II

题目:

给定一个由 n 个节点组成的网络,用 n x n 个邻接矩阵 graph 表示。在节点网络中,只有当 graph[i][j] = 1 时,节点 i 能够直接连接到另一个节点 j

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

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

我们可以从 initial 中删除一个节点并完全移除该节点以及从该节点到任何其他节点的任何连接。

请返回移除后能够使 M(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 中每个整数都不同

思考:

今天的题和昨天的很相似,区别在于:“从 initial 中删除一个节点 = 完全移除该节点以及从该节点到任何其他节点的任何连接

相似的,仍然将图中所有彼此有路径到达的节点们看成一组,如果一组中有至少一个节点初始时被感染,那么这一组所有节点最后都会被感染。

我们要去掉initial中的一个节点和它的所有边之后,使剩下的感染节点最少 ----> 这个节点能且只能凭自己感染的节点最多(1. 通过其他initial节点连接的节点不算  2. 被多个initial节点感染的节点不算)

那么我们的算法步骤如下,数组visited记录每个节点能被多少initial节点凭自己感染("≥0"表示唯一的initial节点索引;"-2"表示有多个initial节点连接);字典sum_dict记录initial节点能且只能凭自己感染的节点数:

1. 遍历initial中的每个节点node。

2. 找到所有和node之间有路径的节点k,并进行判断:1. 若visited[k]为-1,则将visited[k]设为node;2. 若visited[k]为大于等于0的值,说明此前已经有initial节点感染他了,则将visited[k]设为-2.

3. initial中的每个节点node都判断完后,遍历visited数组,若值大于等于0,则说明这个节点只被一个initial节点感染了,将字典sum_dict中该initial节点对应的值加一。

4. 在字典sum_dict中找到值最大的initial节点返回。

代码如下:

from collections import deque


class Solution(object):
    def minMalwareSpread(self, graph, initial):
        """
        :type graph: List[List[int]]
        :type initial: List[int]
        :rtype: int
        """
        # 将互相能到达的节点们视为一个组,(如果initial中有属于这一组的节点)每组的节点数量即为这一个小网络的感染恶意软件的最终节点数
        n = len(graph)
        sum_dict = {}  # 字典sum_dict记录initial节点能且只能凭自己感染的节点数
        visited = [-1] * n  # 数组visited记录节点能被多少initial节点凭自己感染("≥0"表示唯一的initial节点索引;"-2"表示有多个initial节点连接)

        def connectedNodes(graph, initial, node):
            judged = [-1] * n    # 表示在这次遍历中,节点是否已经判断过了
            queue = deque()  # 队列储存待判断相邻节点的节点
            queue.append(node)
            while queue:
                x = queue.popleft()
                for k in range(n):
                    if k == x:  # 跳过当前节点本身
                        continue
                    if judged[k] == -1 and graph[x][k] == 1 and k not in queue and k not in initial:
                        queue.append(k)
                        judged[k] = 1
                        if visited[k] == -1:
                            visited[k] = node
                        elif visited[k] >= 0 and visited[k] != node and graph[x][k] == 1:
                            visited[k] = -2

        for i in initial:
            connectedNodes(graph, initial, i)
            sum_dict[i] = 1

        for j in range(n):
            if visited[j] >= 0:
                sum_dict[visited[j]] += 1

        m = 0
        for key, value in sum_dict.items():    # 在字典sum_dict中找到值最大的initial节点返回
            if value > m:
                m = value
                res = key
            if value == m and key < res:
                res = key
        return res

提交通过,debug了一万年,泪目:

 

最近更新

  1. TCP协议是安全的吗?

    2024-04-22 02:18:01       19 阅读
  2. 阿里云服务器执行yum,一直下载docker-ce-stable失败

    2024-04-22 02:18:01       19 阅读
  3. 【Python教程】压缩PDF文件大小

    2024-04-22 02:18:01       19 阅读
  4. 通过文章id递归查询所有评论(xml)

    2024-04-22 02:18:01       20 阅读

热门阅读

  1. CSS 01

    CSS 01

    2024-04-22 02:18:01      18 阅读
  2. stm32_HAL_串口不定长数据接收发送

    2024-04-22 02:18:01       13 阅读
  3. 升级Linux 4.19至5.10 (失败手稿)

    2024-04-22 02:18:01       20 阅读
  4. yarn的安装与配置(秒懂yarn用法)

    2024-04-22 02:18:01       19 阅读
  5. 0 回归-海上风电出力预测

    2024-04-22 02:18:01       31 阅读
  6. 链表中LinkList L与LinkList *L( & * L.elem L->elem)

    2024-04-22 02:18:01       17 阅读
  7. 常用node.js命令有哪些呢?

    2024-04-22 02:18:01       20 阅读
  8. 面试题汇总

    2024-04-22 02:18:01       14 阅读
  9. Mentor软件安装

    2024-04-22 02:18:01       15 阅读