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 且 0 <= w <= 100。
分析:
本题就是要求解k到其他节点的最短路,然后取这些最短路的最大值。
我们直接使用bellman-ford算法求最短路即可。
bellman-ford算法思想:
n个节点,最多更新n-1遍
每一遍使用所有的m条边进行更新:if(d[v] > d[u] + w(u,v)) d[v] = d[u] + w(u,v)
AC代码:
class Solution {
public:
const int INF=0x3f3f3f3f;
int networkDelayTime(vector<vector<int>>& times, int N, int K) {
int dis[N+5];
for(int i=1;i<=N;i++) dis[i] = INF;
dis[K] = 0;
int num = times.size();
int u,v,w;
for(int i=1;i<N;i++)
{
for(int j=0;j<num;j++)
{
u = times[j][0];
v = times[j][1];
w = times[j][2];
if(dis[v] > dis[u] + w)
{
dis[v] = dis[u] + w;
}
}
}
int maxdis = 0;
for(int i=1;i<=N;i++)
{
maxdis = max(maxdis, dis[i]);
}
if(maxdis == INF) return -1;
else return maxdis;
}
};