Update advent of code

2021/22-1.py

@@ -0,0 +1,83 @@
+def parse_input(s):
+    for line in s.splitlines():
+        state, rest = line.split()
+        x, y, z = rest.split(',')
+        x0, x1 = x[2:].split('..')
+        xrange = range(int(x0), int(x1)+1)
+        y0, y1 = y[2:].split('..')
+        yrange = range(int(y0), int(y1)+1)
+        z0, z1 = z[2:].split('..')
+        zrange = range(int(z0), int(z1)+1)
+
+        yield state, xrange, yrange, zrange
+
+def get_subrange(crange, low, high):
+    c0 = crange[0]
+    c1 = crange[-1]
+    if c1 < low:
+        return []
+    elif c0 > high:
+        return []
+    c0 = max(c0, low)
+    c1 = max(c1, low)
+    c0 = min(c0, high)
+    c1 = min(c1, high)
+    return range(c0, c1+1)
+
+def part1(s):
+    data = list(parse_input(s))
+
+    cubes = {}
+    for idx, item in enumerate(data):
+        state, xr, yr, zr = item
+        for x in get_subrange(xr, -50, 50):
+            for y in get_subrange(yr, -50, 50):
+                for z in get_subrange(zr, -50, 50):
+                    cubes[x,y,z] = state
+
+    answer = sum(1 for s in cubes.values() if s == 'on')
+
+    print(f'The answer to part one is {answer}')
+
+def count_uninterrupted(item, rest):
+    _, xr, yr, zr = item
+    total = len(xr) * len(yr) * len(zr)
+
+    conflicts = []
+    ref_val = 0
+
+    for item in rest:
+        state, xr2, yr2, zr2 = item
+
+        cxr = get_subrange(xr2, xr[0], xr[-1])
+        cyr = get_subrange(yr2, yr[0], yr[-1])
+        czr = get_subrange(zr2, zr[0], zr[-1])
+
+        if len(cxr) == 0 or len(cyr) == 0 or len(czr) == 0:
+            continue
+
+        conflicts.append((state, cxr, cyr, czr))
+        ref_val += len(cxr) * len(cyr) * len(czr)
+
+    for idx, item in enumerate(conflicts):
+        total -= count_uninterrupted(item, conflicts[idx+1:])
+
+    return total
+
+def part2(s):
+    data = list(parse_input(s))
+
+    answer = 0
+
+    for idx, item in enumerate(data):
+        state, xr, yr, zr = item
+        if state == 'off':
+            continue
+        answer += count_uninterrupted(item, data[idx+1:])
+
+    print(f'The answer to part two is {answer}')
+
+with open("./2021/22.input") as input_file:
+    INPUT = input_file.read()
+part1(INPUT)
+part2(INPUT)