mirror of
https://github.com/TheAlgorithms/Python.git
synced 2024-11-24 05:21:09 +00:00
7f04e5cd34
* spelling corrections * review * improved documentation, removed redundant variables, added testing * added type hint * camel case to snake case * spelling fix * review * python --> Python # it is a brand name, not a snake * explicit cast to int * spaces in int list * "!= None" to "is not None" * Update comb_sort.py * various spelling corrections in documentation & several variables naming conventions fix * + char in file name * import dependency - bug fix Co-authored-by: John Law <johnlaw.po@gmail.com>
57 lines
1.6 KiB
Python
57 lines
1.6 KiB
Python
from typing import Dict, List
|
|
|
|
|
|
def printDist(dist, V):
|
|
print("Vertex Distance")
|
|
distances = ("INF" if d == float("inf") else d for d in dist)
|
|
print("\t".join(f"{i}\t{d}" for i, d in enumerate(distances)))
|
|
|
|
|
|
def BellmanFord(graph: List[Dict[str, int]], V: int, E: int, src: int) -> int:
|
|
"""
|
|
Returns shortest paths from a vertex src to all
|
|
other vertices.
|
|
"""
|
|
mdist = [float("inf") for i in range(V)]
|
|
mdist[src] = 0.0
|
|
|
|
for i in range(V - 1):
|
|
for j in range(E):
|
|
u = graph[j]["src"]
|
|
v = graph[j]["dst"]
|
|
w = graph[j]["weight"]
|
|
|
|
if mdist[u] != float("inf") and mdist[u] + w < mdist[v]:
|
|
mdist[v] = mdist[u] + w
|
|
for j in range(E):
|
|
u = graph[j]["src"]
|
|
v = graph[j]["dst"]
|
|
w = graph[j]["weight"]
|
|
|
|
if mdist[u] != float("inf") and mdist[u] + w < mdist[v]:
|
|
print("Negative cycle found. Solution not possible.")
|
|
return
|
|
|
|
printDist(mdist, V)
|
|
return src
|
|
|
|
|
|
if __name__ == "__main__":
|
|
V = int(input("Enter number of vertices: ").strip())
|
|
E = int(input("Enter number of edges: ").strip())
|
|
|
|
graph = [dict() for j in range(E)]
|
|
|
|
for i in range(E):
|
|
graph[i][i] = 0.0
|
|
|
|
for i in range(E):
|
|
print("\nEdge ", i + 1)
|
|
src = int(input("Enter source:").strip())
|
|
dst = int(input("Enter destination:").strip())
|
|
weight = float(input("Enter weight:").strip())
|
|
graph[i] = {"src": src, "dst": dst, "weight": weight}
|
|
|
|
gsrc = int(input("\nEnter shortest path source:").strip())
|
|
BellmanFord(graph, V, E, gsrc)
|