days 15-18 (be careful, crap

2020-12-18 09:20:55 +01:00
parent 1b26f45c1f
commit 0af6bb9594
18 changed files with 1129 additions and 0 deletions
--- a/day16/module.py
+++ b/day16/module.py
@@ -0,0 +1,125 @@
+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()
+
+def p1():
+    inp = get_input(pw)
+    rules_tmp = [x.split(': ') for x in inp[0:20]]
+    my = inp[22].split(',')
+    nearby = [list(map(int, x.split(','))) for x in inp[25:]]
+    rules = {}
+    for r in rules_tmp:
+        rules[r[0]] = list(map(lambda x: list(map(int, x.split('-'))), r[1].split(' or ')))
+    print(rules)
+
+
+    count = 0
+    for n in nearby:
+        valid = True
+        for field in n:
+            fvalid = []
+            for r in rules.values():
+                tvalid = []
+                for rr in r:
+                    tvalid.append(field > rr[0] and field < rr[1])
+                fvalid.append(any(tvalid))
+            if not any(fvalid):
+                count += field
+
+
+
+            pass
+
+    print(count)
+    return inp
+
+
+def p2():
+    inp = get_input(pw)
+    rules_tmp = [x.split(': ') for x in inp[0:20]]
+    my = list(map(int, inp[22].split(',')))
+    nearby = [list(map(int, x.split(','))) for x in inp[25:]]
+    rules = {}
+    for r in rules_tmp:
+        rules[r[0]] = list(map(lambda x: list(map(int, x.split('-'))), r[1].split(' or ')))
+    print(rules)
+
+
+    count = 0
+    vtickets = []
+    print(len(nearby))
+    for n in nearby:
+        valid = []
+        for field in n:
+            fvalid = []
+            for r in rules.values():
+                tvalid = []
+                for rr in r:
+                    tvalid.append(field >= rr[0] and field <= rr[1])
+                fvalid.append(any(tvalid))
+            valid.append(any(fvalid))
+        if all(valid):
+            vtickets.append(n)
+    print(len(vtickets))
+
+
+    print(my)
+    vtickets.append(my)
+    
+    fields = collections.defaultdict(list)
+    for name, r in rules.items():
+        for i in range(len(my)):
+            rvalid = []
+            for n in vtickets:
+                field = n[i]
+                tvalid = []
+                for rr in r:
+                    tvalid.append(field >= rr[0] and field <= rr[1])
+                rvalid.append(any(tvalid))
+            if all(rvalid):
+                fields[i].append(name)
+                print(i, name)
+
+    newfields = copy.copy(fields)
+    print('------')
+    realfields = {}
+    for i in range(len(fields)):
+        i, f = sorted(newfields.items(), key=lambda x: len(x[1]), reverse=False)[0]
+        v = f[0]
+        del newfields[i]
+        realfields[i] = v
+        for i2, f2 in fields.items():
+            if v in  f2:
+                f2.remove(v)
+                fields[i2] = f2
+
+    value = 1
+    for i, v in realfields.items():
+        if v.startswith('departure'):
+            value *= my[i]
+    print(value)
+
+
+
+    return 0
+
+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')
+
+