micro-optimization for Kruskal's algorithm

matteodellamico · Jan 29, 2018 · 8a80782 · 8a80782
1 parent 5125346
commit 8a80782
Showing 1 changed file with 4 additions and 3 deletions.
diff --git a/flexible_clustering/fishdbc.py b/flexible_clustering/fishdbc.py
@@ -177,13 +177,14 @@ def update_mst(self):
         # Kruskal's algorithm
         self._mst_edges = mst_edges = []
         n = len(self.data)
+        needed_edges = n - 1
         uf = UnionFind(n)
-        n_edges = 0
-        while n_edges < n - 1:
+        while needed_edges:
             _, i, j, _ = edge = heapq.heappop(candidate_edges)
             if uf.union(i, j):
                 mst_edges.append(edge)
-                n_edges += 1
+                needed_edges -= 1
+
         new_edges.clear()
 
     def cluster(self, min_cluster_size=None, cluster_selection_method='eom',