Merge pull request luogu-dev#46 from FlierKing/patch-1

Fixed DAG bug
BilyHurington2 · Sep 16, 2018 · 1a320bc · 1a320bc
2 parents 4a52d1d + 9a6a2c7
commit 1a320bc
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/cyaron/graph.py b/cyaron/graph.py
@@ -304,7 +304,7 @@ def DAG(point_count, edge_count, **kwargs):
                 weight_limit[0], weight_limit[1]))
 
         used_edges = set()
-        edge_buf = list(Graph.tree(point_count, weight_limit=weight_gen()).iterate_edges())
+        edge_buf = list(Graph.tree(point_count, weight_limit=weight_limit).iterate_edges())
         graph = Graph(point_count, directed=True)
 
         for edge in edge_buf:
@@ -364,7 +364,7 @@ def UDAG(point_count, edge_count, **kwargs):
                 weight_limit[0], weight_limit[1]))
 
         used_edges = set()
-        graph = Graph.tree(point_count, weight_limit=weight_gen(), directed=False)
+        graph = Graph.tree(point_count, weight_limit=weight_limit, directed=False)
 
         for edge in graph.iterate_edges():
             if not repeated_edges: