题目
一个局域网内有很多台电脑,分别标注为 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
示例1:
输入
4
3
2 1 1
2 3 1
3 4 1
2
输出
2
解决单源加权图最短路径的方法
图详解第四篇:单源最短路径--Dijkstra算法_dijkstra算法 单源最短路径-CSDN博客
图详解第五篇:单源最短路径--Bellman-Ford算法-CSDN博客
代码
n = int(input())
m = int(input())
data = []
for i in range(m):
a, b, temp = map(int, input().split(' '))
data.append([a - 1, b - 1, temp])
k = int(input())
matrix = [[float('inf')] * n for _ in range(n)]
for i in range(n):
matrix[i][i] = 0
for x, y, time in data:
matrix[x][y] = matrix[y][x] = time
for j in range(n):
matrix[x][j] = matrix[j][x] = min(matrix[x][y] + matrix[y][j], matrix[x][j])
matrix[y][j] = matrix[j][y] = min(matrix[y][x] + matrix[x][j], matrix[y][j])
res_time = max(matrix[k])
if res_time == float('inf'):
print(-1)
else:
print(res_time)
仅做学习记录