常见面试算法题-最小传输时延2

■ 题目描述

• 某通信网络中有N个网络结点，用1到N进行标识。网络通过一个有向无环图表示，其中图的边的值表示结点之间的消息传递时延。
• 现给定相连节点之间的时延列表times[i]={u，v，w}，其中u表示源结点，v表示目的结点，w表示u和v之间的消息传递时延。
• 请计算给定源结点到目的结点的最小传输时延，如果目的结点不可达，返回-1。

注：

1. N的取值范围为[1，100];
2. 时延列表times的长度不超过6000，且 1 <= u,v <= N，0 <= w <= 100;

输入描述

1. 输入的第一行为两个正整数，分别表示网络结点的个数N，以及时延列表的长度M，用空格分隔；
2. 接下来的M行为两个结点间的时延列表[u v w];
3. 输入的最后一行为两个正整数，分别表示源结点和目的结点。

输出描述

• 见题目要求

输入

3 3

1 2 11

2 3 13

1 3 50

1 3

输出

24

以下代码为本人原创，可以供大家参考，若有不足之处，感谢指出！！！！

n, m = list(map(int, input().split()))
map1 = {}
for i in range(m):
    ui, vi, wi = list(map(int, input().split()))
    if ui not in map1:
        map1[ui] = {}
    map1[ui].update({vi: wi})
s, e = map(int, input().split())


def dfs(start, end, high):
    min1 = float('inf')
    if start not in map1 or high > n:
        return min1
    next1 = map1[start]
    if end in next1:
        min1 = min(min1, next1[end])
    for k, v in next1.items():
        min1 = min(min1, v+dfs(k, e, high+1))
    return min1


if dfs(s, e, 1) == float('inf'):
    print(-1)
else:
    print(dfs(s, e, 1))