Solve 2016:15 p1-2 "Timing is Everything"

Good TIME to be a Python programmer!

2016-python2/output/day_15.py

@@ -0,0 +1,53 @@
+import re
+
+from output import answer  # , matrix, D, DD, ADJ, ints, mhd, mdbg, vdbg
+
+n = 15
+title = "Timing is Everything"
+
+
+@answer(1, "First time for capsule-giving button press is {}")
+def part_1(presolved):
+    return presolved[0]
+
+
+@answer(2, "With the additional disc, first time for capsule-giving button press is {}")
+def part_2(presolved):
+    return presolved[1]
+
+
+def solve(data):
+    M = [
+        tuple(map(int, re.search(r"(\d+) pos.+ion (\d+)", r).groups()))
+        for r in data.splitlines()
+    ]
+    p1 = wait_and_press(M)
+    p2 = wait_and_press(M + [(11, 0)])
+    return p1, p2
+
+
+def wait_and_press(M):
+    t = 0
+    while True:
+        if all((ti + lp[1]) % lp[0] == 0 for ti, lp in enumerate(M, start=t + 1)):
+            break
+        t += 1
+    return t
+
+
+if __name__ == "__main__":
+    inp = """
+    Disc #1 has 5 positions; at time=0, it is at position 4.
+    Disc #2 has 2 positions; at time=0, it is at position 1.
+    """.strip()
+
+    with open("./input/15.txt", "r") as f:
+        inp = f.read().strip()
+
+    inp = solve(inp)
+
+    a = part_1(inp)
+    b = part_2(inp)
+
+    assert a == 122318
+    assert b == 3208583