From 739416b040965ef61492f07c1d0196fcd850ab54 Mon Sep 17 00:00:00 2001
From: Sebastiano Tronto <sebastiano@tronto.net>
Date: Mon, 8 Dec 2025 18:21:56 +0100
Subject: Small optimization

---
 2025/08/a.py | 5 ++++-
 1 file changed, 4 insertions(+), 1 deletion(-)

(limited to '2025/08/a.py')

diff --git a/2025/08/a.py b/2025/08/a.py
index b829dca..6201d09 100644
--- a/2025/08/a.py
+++ b/2025/08/a.py
@@ -7,7 +7,10 @@ class SetJoin:
 		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])
+		if self._parent[i] != i:
+			self._parent[i] = self.rep(self._parent[i])
+		return self._parent[i]
+
 
 	def join(self, i, j):
 		self._parent[self.rep(i)] = self.rep(j)
-- 
cgit v1.3