kamov/advent-of-code
1
0
Fork 0
Code Issues Pull requests Packages Projects Releases Wiki Activity

Update day21.py

Browse source
This commit is contained in:
kamoshi 2020-12-21 14:03:12 +01:00 committed by GitHub
parent 09ab80a3da
commit 50c5bc115b
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
1 changed files with 18 additions and 5 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

21
2020/Python/day21.py
View file

@ -55,10 +55,23 @@ INGREDIENTS, ALLERGENS, DATA = parse()
print(solve_p1(DATA, INGREDIENTS, ALLERGENS)) print(solve_p1(DATA, INGREDIENTS, ALLERGENS))
def solve_p2(data: list[Tuple[list[str], list[str]]], ingredients: set[str], allergens: set[str]) -> int: def solve_p2(data: list[Tuple[list[str], list[str]]], ingredients: set[str], allergens: set[str]) -> str:
assigned_allergens = sorted([(allergen, set(predict_ingredients(data, allergen))) for allergen in allergens], key=lambda e: len(e[1]))
bijection = []
while len(assigned_allergens[len(assigned_allergens)-1][1]) > 1:
sorted_allergens = assigned_allergens.copy()
first_allergen, first_set = sorted_allergens[0]
bijection.append((first_allergen, first_set))
assigned_allergens = [] assigned_allergens = []
for allergen in allergens: for i in range(1, len(sorted_allergens)):
print(allergen, predict_ingredients(data, allergen)) next_allergen, next_set = sorted_allergens[i]
assigned_allergens.append((next_allergen, next_set.difference(first_set)))
assigned_allergens = sorted(assigned_allergens, key=lambda e: len(e[1]))
bijection += assigned_allergens
result = ",".join([i for (a, i) in sorted(list(map(lambda elem: (elem[0], next(iter(elem[1]))), bijection)), key=lambda e: e[0])])
return result
solve_p2(DATA, INGREDIENTS, ALLERGENS) print(solve_p2(DATA, INGREDIENTS, ALLERGENS))