AcWing 853. 有边数限制的最短路-python3
原题链接
简单
作者:
Luna
,
2019-10-08 11:16:40
,
所有人可见
,
阅读 1064
题目描述
给定一个n个点m条边的有向图,图中可能存在重边和自环, 边权可能为负数。
请你求出从1号点到n号点的最多经过k条边的最短距离,如果无法从1号点走到n号点,输出impossible。
注意:图中可能 存在负权回路 。
输入格式
第一行包含三个整数n,m,k。
接下来m行,每行包含三个整数x,y,z,表示存在一条从点x到点y的有向边,边长为z。
输出格式
输出一个整数,表示从1号点到n号点的最多经过k条边的最短距离。
如果不存在满足条件的路径,则输出“impossible”。
数据范围
1≤n,k≤500,
1≤m≤10000,
任意边长的绝对值不超过10000。
样例
输入样例:
3 3 1
1 2 1
2 3 1
1 3 3
输出样例:
3
python3 代码
import copy
def Bellman_ford():
dist = [float('inf')]*(n+1)
dist[1] = 0
for i in range(k):
backup = copy.copy(dist)
for j in range(m):
start = edges[j]['start']
end = edges[j]['end']
w = edges[j]['w']
dist[end] = min(dist[end],backup[start]+w)
if dist[n] == float('inf'):
return 'impossible'
else:
return dist[n]
n,m,k = map(int,input().split())
edges = []
for i in range(m):
x,y,z = map(int,input().split())
edges.append({'start':x,'end':y,'w':z})
res = Bellman_ford()
print(res)