add 2024/day17.1

2024/17/main.py

@@ -0,0 +1,80 @@
+#!/usr/bin/env python
+# -*- encoding: utf-8; py-indent-offset: 4 -*-
+
+# Author: Sebastian Mark
+# CC-BY-SA (https://creativecommons.org/licenses/by-sa/4.0/deed.de)
+
+# pylint: disable=missing-module-docstring,missing-function-docstring,consider-using-f-string
+
+
+def readinput():
+    with open("input", "r", encoding="utf-8") as file:
+        lines = file.readlines()
+
+        register = {}
+        register["A"] = int(lines[0].split(":")[1].strip())
+        register["B"] = int(lines[1].split(":")[1].strip())
+        register["C"] = int(lines[2].split(":")[1].strip())
+
+        program = list(map(int, lines[4].strip().split(":")[1].split(",")))
+
+    return register, program
+
+
+def get_combo(register: dict, operant: int) -> int:
+    match operant:
+        case 0 | 1 | 2 | 3:
+            return operant
+        case 4:
+            return register["A"]
+        case 5:
+            return register["B"]
+        case 6:
+            return register["C"]
+
+
+def run_programm(register: dict, program: list) -> list:
+    pointer = 0
+    output = []
+    while pointer < len(program):
+        opcode, operant = program[pointer], program[pointer + 1]
+        match opcode:
+            case 0:  # adv
+                combo = get_combo(register, operant)
+                register["A"] //= 2**combo
+            case 1:  # bxl
+                register["B"] ^= operant
+            case 2:  # bst
+                combo = get_combo(register, operant)
+                register["B"] = combo % 8
+            case 3:  # jnz
+                if register["A"] != 0:
+                    pointer = operant
+                    continue
+            case 4:  # bxc
+                register["B"] ^= register["C"]
+            case 5:  # out
+                combo = get_combo(register, operant)
+                output.append(combo % 8)
+            case 6:  # bdv
+                combo = get_combo(register, operant)
+                register["B"] = register["A"] // (2**combo)
+            case 7:  # cdv
+                combo = get_combo(register, operant)
+                register["C"] = register["A"] // (2**combo)
+
+        pointer += 2
+
+    return output
+
+
+def main():
+    register, program = readinput()
+
+    # part 1
+    output = run_programm(register, program)
+    print("Program output: %s" % ",".join(map(str, output)))
+
+
+if __name__ == "__main__":
+    main()