From 5d69f923116a2b251a445c0cb9e6f23551ad686a Mon Sep 17 00:00:00 2001 From: Sebastiano Tronto Date: Mon, 8 Dec 2025 07:29:26 +0100 Subject: Day 8 2025 --- 2025/08/a.py | 30 ++++++++++++++++++++++++++++-- 1 file changed, 28 insertions(+), 2 deletions(-) (limited to '2025/08/a.py') diff --git a/2025/08/a.py b/2025/08/a.py index 5fdb11b..1980414 100644 --- a/2025/08/a.py +++ b/2025/08/a.py @@ -1,6 +1,32 @@ import fileinput +N = 1000 # Change to 10 for test case + with fileinput.input() as lines: - for line in 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 + +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) + +for i in range(N): + j, k = d[i][1], d[i][2] + if findrep(j) != findrep(k): + joinrep(j, k) +sizes = [[0, i] for i in r] +for i in r: + sizes[findrep(i)][0] += 1 +sizes.sort() +print(sizes[-1][0] * sizes[-2][0] * sizes[-3][0]) -- cgit v1.3