题目描述
一个局域网内有很多台电脑,分别标注为 0 ~ N-1 的数字。相连接的电脑距离不一样,所以感染时间不一样,感染时间用 t 表示。
其中网络内一台电脑被病毒感染,求其感染网络内所有的电脑最少需要多长时间。如果最后有电脑不会感染,则返回-1。
给定一个数组 times 表示一台电脑把相邻电脑感染所用的时间。
如图:path[i] = {i, j, t} 表示:电脑 i->j,电脑 i 上的病毒感染 j,需要时间 t。
输入描述
第一行输入一个整数N ,表示局域网内电脑个数 N ,1 ≤ N ≤ 200 ;
第二行输入一个整数M ,表示有 M 条网络连接;
接下来M行 ,每行输入为 i , j , t 。表示电脑 i 感染电脑j 需要时间 t 。(1 ≤ i , j ≤ N)
最后一行为病毒所在的电脑编号。
输出描述
输出最少需要多少时间才能感染全部电脑,如果不存在输出 -1
用例
输入 | 4 3 2 1 1 2 3 1 3 4 1 2 |
---|---|
输出 | 2 |
说明 | 第一个参数:局域网内电脑个数N,1 ≤ N ≤ 200; 第二个参数:总共多少条网络连接 第三个 2 1 1 表示2->1时间为1 第六行:表示病毒最开始所在电脑号2 |
解题思路
Bellman-Ford算法是一种用于在加权图中找到从单个源点到所有其他顶点的最短路径的算法。它能够处理带有负权边的图,这是它与Dijkstra算法的主要区别。然而,如果图中存在负权回路,即一个总权重为负的环路,Bellman-Ford算法可以检测到这种情况。
算法的工作原理如下:
-
初始化距离数组:算法开始时,除了源点(在上面的代码中是变量
K
)的距离被初始化为0以外,所有顶点的距离都被设置为无穷大(在上面的代码中是INF
)。 -
松弛操作:算法会进行
N-1
次迭代,其中N
是图中顶点的数量。在每次迭代中,算法会遍历所有的边,并尝试更新每条边的目标顶点的距离。如果通过当前边到达目标顶点的距离小于已知的最短距离,则更新该顶点的最短距离。这个过程称为松弛操作。 -
检测负权回路:在
N-1
次迭代之后,算法会再次遍历所有的边,检查是否还能进行松弛操作。如果可以,这意味着图中存在负权回路,因为最短路径应该已经在前面的N-1
次迭代中被确定下来。
在下面的代码中,networkDelayTime
函数实现了Bellman-Ford算法:
times
数组包