Leetcode算法——2039、网络空闲的时刻

题目

题目链接

思路

广搜，广搜思路与树的层序遍历一样，广搜中的数据结构最好使用双向队列deque（重点），因为广搜频繁进行头部元素的弹出，而使用列表的话弹出头部元素是O(n)操作，而双向队列是O(1)操作。该题使用列表会超时。
下面计算最近的空闲时间思路比较简单，可以参考官方答案

Python3解法

def networkBecomesIdle(self, edges: List[List[int]], patience: List[int]) -> int:
    """
    q必须是双向队列deque,使用列表的话弹出头部元素是O(n)操作，而双向队列是O(1)操作
    """
    l = len(patience)
    q, dist, vis, m, res = deque([0]), 1, [False] * l, [[] for i in range(l)], -1
    vis[0] = True
    for left, right in edges:
        m[left].append(right)
        m[right].append(left)
    while q:
        for _ in range(len(q)):
            cur = q.popleft()
            for node in m[cur]:
                if vis[node]:   continue
                vis[node] = True
                q.append(node)
                res = max(res, (dist * 2 - 1) // patience[node] * patience[node] + dist * 2 + 1)
        dist += 1
    return res