TheAlgorithms/Python
1
0
Fork 0
Code Issues Pull Requests Actions 2 Packages Projects Releases Wiki Activity

Travis CI: Add pytest --doctest-modules graphs (#1018)

Browse Source
This commit is contained in:
cclauss
2019-07-17 06:07:25 +02:00
committed by GitHub
parent 267b5eff40
commit 7cdda931fd
9 changed files with 145 additions and 132 deletions
Download Patch File Download Diff File Expand all files Collapse all files

19
graphs/minimum_spanning_tree_prims.py
View File

@@ -100,12 +100,13 @@ def PrimsAlgorithm(l):
Nbr_TV[ v[0] ] = vertex
return TreeEdges
# < --------- Prims Algorithm --------- >
n = int(input("Enter number of vertices: "))
e = int(input("Enter number of edges: "))
adjlist = defaultdict(list)
for x in range(e):
l = [int(x) for x in input().split()]
adjlist[l[0]].append([ l[1], l[2] ])
adjlist[l[1]].append([ l[0], l[2] ])
print(PrimsAlgorithm(adjlist))
if __name__ == "__main__":
# < --------- Prims Algorithm --------- >
n = int(input("Enter number of vertices: ").strip())
e = int(input("Enter number of edges: ").strip())
adjlist = defaultdict(list)
for x in range(e):
l = [int(x) for x in input().strip().split()]
adjlist[l[0]].append([ l[1], l[2] ])
adjlist[l[1]].append([ l[0], l[2] ])
print(PrimsAlgorithm(adjlist))