华为OD机试 Python【局域网病毒感染时间计算】

题目

一个局域网内有很多台电脑，分别标注为 0 ~ N-1 的数字。相连接的电脑距离不一样，所以感染时间不一样，感染时间用 t 表示。其中网络内一台电脑被病毒感染，求其感染网络内所有的电脑最少需要多长时间。如果最后有电脑不会感染，则返回-1。给定一个数组 times 表示一台电脑把相邻电脑感染所用的时间。

如图：path[i] = {i, j, t} 表示：电脑 i->j，电脑 i 上的病毒感染 j，需要时间 t。

输入描述
4
3
2 1 1
2 3 1
3 4 1
2
输出描述
2

解题思路

该问题可以抽象为带权图中的最短路径问题。使用Dijkstra算法来计算从某一台电脑开始感染其它所有电脑所需的最短时间。如果存在电脑无法被感染，则返回-1。

代码

import heapq