feat: 2024 day 17

Author: Programator2

2024/17a.py

@@ -0,0 +1,56 @@
+from aoc import *
+
+
+def main(infi: str):
+    inp = filerstrip(infi)
+    registers, program = inp.split('\n\n')
+    a, b, c = map(
+        lambda x: int(x[len('Register A: ') :]), registers.split('\n')
+    )
+
+    def combo(op):
+        if 0 <= op <= 3:
+            return op
+        elif op == 4:
+            return a
+        elif op == 5:
+            return b
+        elif op == 6:
+            return c
+        raise RuntimeError('combo')
+
+    prog = list(map(int, program[len('Program: ') :].split(',')))
+    ip = 0
+    out = []
+    while 0 <= ip < len(prog):
+        if prog[ip] == 0:
+            a = a // (2 ** combo(prog[ip + 1]))
+        elif prog[ip] == 1:
+            b = b ^ prog[ip + 1]
+        elif prog[ip] == 2:
+            b = combo(prog[ip + 1]) % 8
+        elif prog[ip] == 3:
+            if a:
+                ip = prog[ip + 1]
+                continue
+        elif prog[ip] == 4:
+            b = b ^ c
+        elif prog[ip] == 5:
+            out.append(combo(prog[ip + 1]) % 8)
+            # print(out[-1])
+        elif prog[ip] == 6:
+            b = a // (2 ** combo(prog[ip + 1]))
+        elif prog[ip] == 7:
+            c = a // (2 ** combo(prog[ip + 1]))
+        ip += 2
+    return ','.join(map(str, out))
+
+
+DAY = 17
+FILE_TEST = f"{DAY}_testa.txt"
+# FILE_TEST = f"{DAY}_testb.txt"
+FILE_EXP = f"{DAY}_exp.txt"
+FILE = f"{DAY}.txt"
+# test_and_submit(main, FILE_TEST, FILE_EXP, FILE, DAY)
+# print(main(FILE_TEST))
+print(main(FILE))

2024/17b.py

@@ -0,0 +1,46 @@
+from aoc import *
+
+
+def solve(ae, out):
+    if not out:
+        return ae
+    # For the first iteration (lowest search depth), the range should be (1, 8).
+    # But it doesn't matter, since the result is found anyway.
+    for rem in range(0, 8):
+        astart = ae * 8 + rem
+        b1 = rem
+        b2 = b1 ^ 5
+        c1 = astart // 2**b2
+        b3 = b2 ^ c1
+        b4 = b3 ^ 6
+        oute = b4 % 8
+        if oute == out[0]:
+            if (ret := solve(astart, out[1:])) is not None:
+                return ret
+
+
+def main(infi: str):
+    out = list(
+        reversed(
+            list(
+                map(
+                    int,
+                    filerstrip(infi)
+                    .split('\n\n')[1][len('Program: ') :]
+                    .split(','),
+                )
+            )
+        )
+    )
+
+    return solve(0, out)
+
+
+DAY = 17
+FILE_TEST = f"{DAY}_testa.txt"
+# FILE_TEST = f"{DAY}_testb.txt"
+FILE_EXP = f"{DAY}_exp.txt"
+FILE = f"{DAY}.txt"
+# test_and_submit(main, FILE_TEST, FILE_EXP, FILE, DAY)
+# print(main(FILE_TEST))
+print(main(FILE))

README.org

@@ -4,6 +4,7 @@
 * Personal Stats 2024
 | Day |     Time | Rank | Score |     Time |  Rank | Score |
 |-----+----------+------+-------+----------+-------+-------|
+|  17 | 00:27:08 | 1748 |     0 | 07:06:10 |  5688 |     0 |
 |  15 | 00:32:17 | 2397 |     0 | 02:59:49 |  4379 |     0 |
 |  14 | 00:19:09 | 1902 |     0 | 01:13:46 |  3551 |     0 |
 |  13 | 00:39:23 | 4308 |     0 | 12:34:52 | 21706 |     0 |