LeetCode743. 网络延迟时间(python,bfs)

1. 题目

有 N 个网络节点，标记为 1 到 N。

给定一个列表 times，表示信号经过有向边的传递时间。 times[i] = (u, v, w)，其中 u 是源节点，v 是目标节点， w 是一个信号从源节点传递到目标节点的时间。

现在，我们从某个节点 K 发出一个信号。需要多久才能使所有节点都收到信号？如果不能使所有节点收到信号，返回 -1。

示例：

输入：times = [[2,1,1],[2,3,1],[3,4,1]], N = 4, K = 2
输出：2

注意:

N 的范围在 [1, 100] 之间。
K 的范围在 [1, N] 之间。
times 的长度在 [1, 6000] 之间。
所有的边 times[i] = (u, v, w) 都有 1 <= u, v <= N 且 0 <= w <= 100。

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

2. 解答

还是常规bfs问题。

class Solution:
    def networkDelayTime(self, times: List[List[int]], N: int, K: int) -> int:
        #bfs

        queue = deque()
        queue.append((K, 0))
        visited = [K]
        res_dict = {}
        res_dict[K] = 0
        while len(queue)>0: 
            x, t = queue.popleft()
            for point in times:
                if point[0] == x:
                    nx = point[1]
                    if nx not in visited:
                        visited.append(nx)
                        res_dict[nx] = t+point[2]
                        queue.append((nx,t+point[2]))
                    else:
                        if t+point[2] < res_dict[nx]:
                            res_dict[nx] = t+point[2]
                            queue.append((nx,t+point[2]))
        if len(visited) != N:
            return -1
        return max(res_dict.values())