阿里算法题-2019届

光明小学的小朋友们要举行一年一度的接力跑大赛了，可是小朋友们却遇到了一个难题：设计接力跑大赛的线路，你能帮助他们完成这项工做么？
光明小学能够抽象成一张有N个节点的图，每两点间都有一条道路相连。光明小学的每一个班都有M个学生，因此你要为他们设计出一条刚好通过M条边的路径。
光明小学的小朋友们但愿全盘考虑全部的因素，因此你须要把任意两点间通过M条边的最短路径的距离输出出来以供参考。

你须要设计这样一个函数：
res[][] Solve( N, M, map[][]);
注意：map必然是N * N的二维数组，且map[i][j] == map[j][i]，map[i][i] == 0，-1e8 <= map[i][j] <= 1e8。（道路所有是无向边，无自环）2 <= N <= 100, 2 <= M <= 1e6。要求时间复杂度控制在O(N^3*log(M))。

map数组表示了一张稠密图，其中任意两个不一样节点i,j间都有一条边，边的长度为map[i][j]。N表示其中的节点数。
你要返回的数组也必然是一个N * N的二维数组，表示从i出发走到j，通过M条边的最短路径
你的路径中应考虑包含重复边的状况。

输入样例：

3
2
3 3 
0 2 3
2 0 1
3 1 0

输出样例：

[[4, 4, 3],

[4, 2, 5],

[3, 5, 2]]

 

 1 if __name__=='__main__':
 2     #   未添加边界值的检查
 3     n = int(raw_input())
 4     m = int(raw_input())
 5     x = raw_input()
 6     map = []
 7     for i in range(n):
 8         line = [int(x) for x in raw_input().split()]
 9         map.append(list(line))
10     
11     pre_mat = [list(x) for x in map]
12     c_mat = [list(x) for x in map]
13     
14     for _ in range(m-1):
15         for start in range(n):
16             for end in range(start, n):
17                 paths = [map[start][mid]+pre_mat[mid][end] for mid in range(n) if mid!=start and pre_mat[mid][end]!=0]
18                 c_mat[start][end]=c_mat[end][start]=min(paths)
19         pre_mat = [list(x) for x in c_mat]
20     
21     print c_mat

 

30分钟,凉凉.....