阿里算法题-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分钟,凉凉.....

posted @ 2018-07-25 17:09  rw_rongwei  阅读(217)  评论(0编辑  收藏  举报