Added deep rerooting while finding cheapest tree

ipgmvq · Apr 3, 2023 · 18e5c11 · 18e5c11
1 parent 17394d4
commit 18e5c11
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/nodeweightedbudget/core.py b/nodeweightedbudget/core.py
@@ -100,6 +100,8 @@ def find_short_path(graph, adj_matrix, depth, previous, current):
             elif graph.vertices[i].distance > distance:
                 graph.vertices[i].distance = distance
                 graph.vertices[i].parent = current
+                graph.find_short_path(
+                    adj_matrix=adj_matrix, depth=depth+1, previous=current, current=i)
         for i, v in graph.vertices.items():
             if v.parent is not None:
                 graph.vertices[v.parent].children.add(i)