days 15-18 (be careful, crap

day15/module.py

@@ -0,0 +1,64 @@
+from aoc.input import get_input
+import copy
+import itertools
+import time
+import collections
+import re
+from aoc.partselector import part_one, part_two
+import functools
+
+
+def pw(line):
+    return line.strip().split(",")
+
+
+def p1():
+    inp = get_input(pw)
+    nmbrs = []
+    for sample in inp:
+        print(sample)
+    for i in inp:
+        nmbrs.append(int(i))
+    last = inp[-1]
+    for i in range(len(inp), 30000000):
+        if last not in nmbrs or len(list(filter(lambda x: x == last, nmbrs))) == 1:
+            number = 0
+        else:
+            pos = len(nmbrs) - nmbrs[-2::-1].index(last) - 2
+            opos = len(nmbrs) - 1
+            number = opos - pos
+
+        print(i + 1, ":", number, number - i)
+        nmbrs.append(number)
+        last = number
+
+    return inp
+
+
+def p2():
+    inp = get_input(pw)
+    nmbrs = {int(x): y + 1 for y, x in enumerate(inp[:-1])}
+    last = int(inp[-1])
+
+    for i in range(len(nmbrs) + 1, 30000000):
+        if last not in nmbrs:
+            nmbrs[last] = i
+            last = 0
+        else:
+            prev, nmbrs[last] = nmbrs[last], i
+            last = i - prev
+
+    print(last)
+
+
+result1 = None
+if part_one():
+    start = time.time()
+    result1 = p1()
+    print(round(1000 * (time.time() - start), 2), "ms")
+
+
+if part_two():
+    start = time.time()
+    p2()
+    print(round(1000 * (time.time() - start), 2), "ms")