diff --git a/graphs/graphs_floyd_warshall.py b/graphs/graphs_floyd_warshall.py index 629da37af..c8d8315a2 100644 --- a/graphs/graphs_floyd_warshall.py +++ b/graphs/graphs_floyd_warshall.py @@ -9,10 +9,11 @@ def _print_dist(dist, v): print("The shortest path matrix using Floyd Warshall algorithm\n") for i in range(v): for j in range(v): + end_char = "" if j == v - 1 else "\t" if dist[i][j] != float("inf"): - print(int(dist[i][j]), end=" ") + print(int(dist[i][j]), end=end_char) else: - print("INF", end=" ") + print("INF", end=end_char) print() @@ -47,9 +48,9 @@ def floyd_warshall(graph, v): >>> dist, _ = floyd_warshall(graph, 3) The shortest path matrix using Floyd Warshall algorithm - 0 3 INF - 2 0 INF - 9 7 0 + 0 3 INF + 2 0 INF + 9 7 0 >>> dist == expected True """