You cannot select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
39 lines
1.1 KiB
Python
39 lines
1.1 KiB
Python
"""Solution for 2020/21"""
|
|
|
|
from functools import reduce
|
|
|
|
foods = []
|
|
all_ingredients = set()
|
|
all_allergenes = set()
|
|
with open('input', 'r') as f:
|
|
for line in f.readlines():
|
|
[ing, alg] = line.strip(' \n)').split(' (contains ')
|
|
ing = set(ing.split(' '))
|
|
alg = set(alg.split(', '))
|
|
all_ingredients = all_ingredients.union(ing)
|
|
all_allergenes = all_allergenes.union(alg)
|
|
foods.append((ing, alg))
|
|
|
|
allergeneMap = {}
|
|
for ingredients, allergenes in foods:
|
|
for allergene in allergenes:
|
|
if allergene not in allergeneMap:
|
|
allergeneMap[allergene] = ingredients
|
|
else:
|
|
allergeneMap[allergene] = allergeneMap[allergene].intersection(
|
|
ingredients)
|
|
print(f'{allergeneMap=}')
|
|
|
|
allergenic = reduce(lambda x, y: x.union(y), allergeneMap.values())
|
|
|
|
unallergenic = all_ingredients.difference(allergenic)
|
|
print(f'{allergenic=}')
|
|
print(f'{unallergenic=}')
|
|
|
|
solution = 0
|
|
for ingredients, _ in foods:
|
|
solution += len(ingredients.difference(allergenic))
|
|
print(
|
|
f'{ingredients} - {allergenic} = {len(ingredients.difference(allergenic))}')
|
|
print(solution)
|