From 2524a8856bb743139a107d59f14798f5e87231ce Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Gustav=20S=C3=B6rn=C3=A4s?= Date: Mon, 21 Dec 2020 11:12:34 +0100 Subject: d21 cleanup --- 20/py/d21.py | 9 +++------ 1 file changed, 3 insertions(+), 6 deletions(-) diff --git a/20/py/d21.py b/20/py/d21.py index 82f5b00..0ed0448 100644 --- a/20/py/d21.py +++ b/20/py/d21.py @@ -13,17 +13,14 @@ def allergens(line): def pt1(_in): - all_ingredients = [] - for line in _in: - all_ingredients.extend(ingredients(line)) + all_ingredients = list(itertools.chain.from_iterable(ingredients(line) for line in _in)) might_be = defaultdict(lambda: set(all_ingredients)) for line in _in: for alg in allergens(line): might_be[alg] &= set(ingredients(line)) - might_be = set.union(*might_be.values()) - defo_not = set(all_ingredients) - might_be - return sum(all_ingredients.count(ing) for ing in defo_not) + return sum(all_ingredients.count(ing) + for ing in set(all_ingredients) - set.union(*might_be.values())) def pt2(_in): -- cgit v1.2.1