743. 网络延迟时间

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

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

现在,我们向当前的节点 K 发送了一个信号。需要多久才能使所有节点都收到信号?如果不能使所有节点收到信号,返回 -1。

注意:

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

import queue 
class Solution:
    
    def networkDelayTime(self, times, N, K):
        """
        :type times: List[List[int]]
        :type N: int
        :type K: int
        :rtype: int
        """
        dict={i:{} for i in range(1,N+1)}

        for i in times:
            u=i[0]
            v=i[1]
            w=i[2]
            dict[u][v]=w

        time_dict={i:-1 for i in range(1,N+1)}
        #start at k
        time_dict[K]=0
        q=queue.Queue()
        q.put(K)
        while not q.empty():
            x=q.get()
            sons_cost=tuple(dict[x].items())
            for (son,cost) in sons_cost:
                if time_dict[son]==-1:
                    time_dict[son]=time_dict[x]+cost
                    q.put(son)
                else:
                    if time_dict[son]>time_dict[x]+cost:
                        time_dict[son]=time_dict[x]+cost
                        q.put(son)
        if -1 not in list(time_dict.values()):
            return max(list(time_dict.values()))
        else:
            return -1
                    
        

你可能感兴趣的:(743. 网络延迟时间)