2024. 1华为od机试C卷【电脑病毒感染】Python

题目

一个局域网内有很多台电脑，分别标注为 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)

仅做学习记录