有 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
网友评论