🐜 Study: 전력망을 둘로 나누기 (#78)

Author: ing-eoking

전력망을 둘로 나누기.py

@@ -0,0 +1,10 @@
+def solution(n, wires):
+    answer = []
+    find = lambda x: x if x == root[x] else find(root[x])
+    for i in range(len(wires)):
+        nw = wires[:i] + wires[i+1:]
+        root = list(range(n))
+        for a, b in nw:
+            if (x := find(a - 1)) != (y := find(b - 1)): root[y] = x
+        answer.append([find(i) for i in root].count(find(root[0])))
+    return min([abs(n - 2 * x) for x in answer])