oysteikt-skolearbeid/MA0301
oysteikt-skolearbeid
/
MA0301
1
0
Fork 0
Code Issues Pull Requests Activity

Add error check to kruskals

This commit is contained in:
Oystein Kristoffer Tveit 2021-05-17 23:37:23 +02:00
parent bc1609c549
commit b526c648c4
1 changed files with 5 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
exam_template/python/Graph.py
View File

@ -2,7 +2,7 @@ from sys import argv
from pathlib import Path from pathlib import Path
from math import sin, cos, pi from math import sin, cos, pi
from common import printc, replaceContent from common import printc, printerr, replaceContent
class Matrix: class Matrix:
""" Adjacency matrix which supports 0 and 1 """ """ Adjacency matrix which supports 0 and 1 """
@ -127,6 +127,10 @@ class Graph:
edges = [] edges = []
connected_subgraphs = [set(v) for v in self.nodes] connected_subgraphs = [set(v) for v in self.nodes]
if not all(n in ''.join(x+y for x,y in self.edges) for n in self.nodes):
printerr('Not all nodes are connected. There is no spanning tree to be made')
return
def find_subgraph_containing(v): def find_subgraph_containing(v):
return [x for x in connected_subgraphs if v in x][0] return [x for x in connected_subgraphs if v in x][0]