@@ -45,35 +45,35 @@ def min_dist(mdist, vset, v):
4545
4646
4747def dijkstra (graph , v , src ):
48- """
49- Runs Dijkstra's algorithm and prints distances.
50- >>> g = [
51- ... [0.0, 5.0, float('inf'), 10.0],
52- ... [float('inf'), 0.0, 3.0, float('inf')],
53- ... [float('inf'), float('inf'), 0.0, 1.0],
54- ... [float('inf'), float('inf'), float('inf'), 0.0],
55- ... ]
56- >>> dijkstra(g, 4, 0)
57- Vertex Distance
58- 0 0
59- 1 5
60- 2 8
61- 3 9
62- >>> g2 = [
63- ... [0.0, float('inf')],
64- ... [float('inf'), 0.0],
65- ... ]
66- >>> dijkstra(g2, 2, 0)
67- Vertex Distance
68- 0 0
69- 1 INF
70- >>> dijkstra([[0.0]], 1, 0)
71- Vertex Distance
72- 0 0
73- """
74- mdist = [float ("inf" ) for _ in range (v )]
75- vset = [False for _ in range (v )]
76- mdist [src ] = 0.0
48+ """
49+ Runs Dijkstra's algorithm and prints distances.
50+ >>> g = [
51+ ... [0.0, 5.0, float('inf'), 10.0],
52+ ... [float('inf'), 0.0, 3.0, float('inf')],
53+ ... [float('inf'), float('inf'), 0.0, 1.0],
54+ ... [float('inf'), float('inf'), float('inf'), 0.0],
55+ ... ]
56+ >>> dijkstra(g, 4, 0)
57+ Vertex Distance
58+ 0 0
59+ 1 5
60+ 2 8
61+ 3 9
62+ >>> g2 = [
63+ ... [0.0, float('inf')],
64+ ... [float('inf'), 0.0],
65+ ... ]
66+ >>> dijkstra(g2, 2, 0)
67+ Vertex Distance
68+ 0 0
69+ 1 INF
70+ >>> dijkstra([[0.0]], 1, 0)
71+ Vertex Distance
72+ 0 0
73+ """
74+ mdist = [float ("inf" ) for _ in range (v )]
75+ vset = [False for _ in range (v )]
76+ mdist [src ] = 0.0
7777
7878 for _ in range (v - 1 ):
7979 u = min_dist (mdist , vset , v )
0 commit comments