changing end character in print statement

This commit is contained in:
Ajmera, Mahita SI/HZR-IDSA 2024-10-17 11:57:41 +02:00
parent 3e623c1476
commit be16a1ef9b

View File

@ -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
<BLANKLINE>
0 3 INF
2 0 INF
9 7 0
0 3 INF
2 0 INF
9 7 0
>>> dist == expected
True
"""