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