Refactor

1 files changed, 14 insertions, 12 deletions

diff --git a/2025/08/b.py b/2025/08/b.py
index 4a6d610..0fac672 100644
--- a/2025/08/b.py
+++ b/2025/08/b.py
@@ -1,26 +1,28 @@
 import fileinput
 
+class SetJoin:
+        def __init__(self, n):
+                self._parent = [i for i in range(n)]
+
+        def rep(self, i):
+                return i if self._parent[i] == i else self.rep(self._parent[i])
+
+        def join(self, i, j):
+                self._parent[self.rep(i)] = self.rep(j)
+
 with fileinput.input() as lines:
         pts = [tuple(int(x) for x in line[:-1].split(',')) for line in lines]
 
-r = range(len(pts))
-
 def dist(p, q):
         return (p[0]-q[0])**2 + (p[1]-q[1])**2 + (p[2]-q[2])**2
 
+r = range(len(pts))
 d = sorted([(dist(pts[i], pts[j]), i, j) for i in r for j in r if j > i])
 
-rep = [i for i in r]
-
-def findrep(i):
-        return i if rep[i] == i else findrep(rep[i])
-
-def joinrep(i, j):
-        rep[findrep(i)] = findrep(j)
-
+sj = SetJoin(len(pts))
 for _, j, k in d:
-        if findrep(j) != findrep(k):
-                joinrep(j, k)
+        if sj.rep(j) != sj.rep(k):
+                sj.join(j, k)
                 sol = pts[j][0] * pts[k][0]
 
 print(sol)