854. Floyd求最短路（python3）

给定一个 n𝑛 个点 m𝑚 条边的有向图，图中可能存在重边和自环，边权可能为负数。

再给定 k𝑘 个询问，每个询问包含两个整数 x𝑥 和 y𝑦，表示查询从点 x𝑥 到点 y𝑦 的最短距离，如果路径不存在，则输出 impossible。

数据保证图中不存在负权回路。

输入格式

第一行包含三个整数 n,m,k𝑛,𝑚,𝑘。

接下来 m𝑚 行，每行包含三个整数 x,y,z𝑥,𝑦,𝑧，表示存在一条从点 x𝑥 到点 y𝑦 的有向边，边长为 z𝑧。

接下来 k𝑘 行，每行包含两个整数 x,y𝑥,𝑦，表示询问点 x𝑥 到点 y𝑦 的最短距离。

输出格式

共 k𝑘 行，每行输出一个整数，表示询问的结果，若询问两点间不存在路径，则输出 impossible。

数据范围

1≤n≤2001≤𝑛≤200,
1≤k≤n21≤𝑘≤𝑛2
1≤m≤200001≤𝑚≤20000,
图中涉及边长绝对值均不超过 1000010000。

输入样例：

3 3 2
1 2 1
2 3 2
1 3 1
2 1
1 3

输出样例：

impossible
1

代码 

n, m, k = map(int, input().split())
d = [[float('inf')]*(n+1) for _ in range(n+1)]
# 初始化距离矩阵，每个点到自身距离为0
for i in range(1, n+1):
    for j in range(1, n+1):
        if i == j :
            d[i][j] = 0
# 读入边的信息
for _ in range(m):
    u, v, w = map(int, input().split())
    d[u][v] = min(d[u][v], w) # 只存最小的边

def floyd():
    for k in range(1, n+1): # 对于中间节点k
        for i in range(1, n+1): # 起始节点
            for j in range(1, n+1):
                # 更新节点距离
                d[i][j] = min(d[i][j], d[i][k]+d[k][j])
floyd()

for _ in range(k):
    u, v = map(int, input().split())
    if d[u][v] >= float('inf') /2:
        print('impossible')
    else:
        print(d[u][v])