kobo

y2021/p22.py

+import re
+
+
+def intersect(cube, other):
+    return not (
+        cube[1] <= other[0] or
+        cube[0] >= other[1] or
+        cube[3] <= other[2] or
+        cube[2] >= other[3] or
+        cube[5] <= other[4] or
+        cube[4] >= other[5]
+    )
+
+
+def insert(cube, val):
+    to_break = {other for other in seen if intersect(cube, other)}
+    for cu in to_break:
+        seen.discard(cu)
+
+    xAs, xBs, yAs, yBs, zAs, zBs = zip(*{cube} | to_break)
+    xs = sorted(set(xAs + xBs))
+    ys = sorted(set(yAs + yBs))
+    zs = sorted(set(zAs + zBs))
+    N = 0
+    for i in range(len(xs) - 1):
+        for j in range(len(ys) - 1):
+            for k in range(len(zs) - 1):
+                new = (xs[i], xs[i + 1], ys[j], ys[j + 1], zs[k], zs[k + 1])
+
+                if val == 'off' and intersect(new, cube):
+                    seen.discard(new)
+
+                elif val == 'on' and intersect(new, cube):
+                    seen.add(new)
+
+                elif any(intersect(old, new) for old in to_break):
+                    seen.add(new)
+
+
+def volume(xA, xB, yA, yB, zA, zB):
+    return abs(xB - xA) * abs(yB - yA) * abs(zB - zA)
+
+
+text = open(0).read()
+cubes = {}
+for line in text.splitlines():
+    val = line.split()[0]
+    xA, xB, yA, yB, zA, zB = [int(n) for n in re.findall(r'-?\d+', line)]
+    cubes[xA, xB + 1, yA, yB + 1, zA, zB + 1] = val
+
+
+seen = set()
+for i, (cube, val) in enumerate(cubes.items(), 1):
+    insert(cube, val)
+    if i == 20:
+        print(sum(volume(*cu) for cu in seen))
+print(sum(volume(*cu) for cu in seen))