day21

day21/module.py

@@ -0,0 +1,79 @@
+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)
+    lst = []
+    ingredients = set()
+    lingredients = []
+    idict = collections.defaultdict(list)
+    for sample in inp:
+        i, a = sample.split('(contains ')
+        a = a[:-1].strip()
+        lst.append((i.strip().split(' '), a.split(', ')))
+        is_ = i.strip().split(' ')
+        lingredients.extend(is_)
+        for x in is_:
+            ingredients.add(x)
+        for a2 in a.split(', '):
+            idict[a2].append(is_)
+
+    allergens = set()
+    hazzard = collections.defaultdict(list)
+    for allergen, t in idict.items():
+        v = set(t[0])
+        for tt in t[1:]:
+            v = v.intersection(tt)
+        for x in v:
+            allergens.add(x)
+        for x in v:
+            hazzard[x].append(allergen)
+    count = 0
+    for l in lingredients:
+        if l not in allergens:
+            count += 1
+    print(count)
+    return hazzard
+
+
+def p2(hazzard):
+    while any(map(lambda x: len(x)>1, hazzard.values())):
+        hazzard2 = {}
+        for i, a in hazzard.items():
+            for a2 in filter(lambda x: len(x) == 1, hazzard.values()):
+                if a == a2:
+                    continue
+                v = set(a).symmetric_difference(a2) & set(a)
+                if len(v) <= len(hazzard[i]) and len(v) > 0:
+                    if i in hazzard2:
+                        hazzard2[i] = list(v.intersection(hazzard2[i]))
+                    else:
+                        hazzard2[i] = list(v)
+        hazzard = hazzard2
+    print(','.join([x[0] for x in sorted(hazzard.items(), key=lambda x: x[1])]))
+
+
+    return 0
+
+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(result1)
+    print(round(1000*(time.time() - start), 2), 'ms')
+
+