65 lines
2.4 KiB
Python
65 lines
2.4 KiB
Python
|
from typing import Tuple
|
||
|
|
|
||
|
|
|
||
|
|
def parse() -> Tuple[set[str], set[str], list[Tuple[list[str], list[str]]]]:
|
||
|
|
ingredient_set = set()
|
||
|
|
allergen_set = set()
|
||
|
|
output = []
|
||
|
|
with open("input.txt") as file:
|
||
|
|
for line in file:
|
||
|
|
ingredients, allergens = line.rstrip().split(" (contains ")
|
||
|
|
ingredients_list = list(map(str.strip, ingredients.split(" ")))
|
||
|
|
allergens_list = list(map(str.strip, allergens[:-1].split(",")))
|
||
|
|
[ingredient_set.add(ingredient) for ingredient in ingredients_list]
|
||
|
|
[allergen_set.add(allergen) for allergen in allergens_list]
|
||
|
|
output.append((ingredients_list, allergens_list))
|
||
|
|
return ingredient_set, allergen_set, output
|
||
|
|
|
||
|
|
|
||
|
|
def predict_ingredients(data: list[Tuple[list[str], list[str]]], allergen: str) -> list[str]:
|
||
|
|
counted_occurrences = {}
|
||
|
|
for ingredients, allergens in data:
|
||
|
|
if allergen in allergens:
|
||
|
|
for ingredient in ingredients:
|
||
|
|
if ingredient not in counted_occurrences:
|
||
|
|
counted_occurrences[ingredient] = 0
|
||
|
|
counted_occurrences[ingredient] += 1
|
||
|
|
|
||
|
|
found_ingredients = set()
|
||
|
|
for ingredients, allergens in data:
|
||
|
|
if allergen in allergens:
|
||
|
|
max_occurrences = max([val if key in ingredients else 0 for (key, val) in counted_occurrences.items()])
|
||
|
|
for ingredient in ingredients:
|
||
|
|
if counted_occurrences[ingredient] == max_occurrences:
|
||
|
|
found_ingredients.add(ingredient)
|
||
|
|
|
||
|
|
return list(found_ingredients)
|
||
|
|
|
||
|
|
|
||
|
|
def solve_p1(data: list[Tuple[list[str], list[str]]], ingredients: set[str], allergens: set[str]) -> int:
|
||
|
|
probable_ingredients = set()
|
||
|
|
for allergen in allergens:
|
||
|
|
for found in predict_ingredients(data, allergen):
|
||
|
|
probable_ingredients.add(found)
|
||
|
|
|
||
|
|
safe = ingredients.difference(probable_ingredients)
|
||
|
|
counted = 0
|
||
|
|
for ingredients, _ in data:
|
||
|
|
for ingredient in ingredients:
|
||
|
|
if ingredient in safe:
|
||
|
|
counted += 1
|
||
|
|
return counted
|
||
|
|
|
||
|
|
|
||
|
|
INGREDIENTS, ALLERGENS, DATA = parse()
|
||
|
|
print(solve_p1(DATA, INGREDIENTS, ALLERGENS))
|
||
|
|
|
||
|
|
|
||
|
|
def solve_p2(data: list[Tuple[list[str], list[str]]], ingredients: set[str], allergens: set[str]) -> int:
|
||
|
|
assigned_allergens = []
|
||
|
|
for allergen in allergens:
|
||
|
|
print(allergen, predict_ingredients(data, allergen))
|
||
|
|
|
||
|
|
|
||
|
|
solve_p2(DATA, INGREDIENTS, ALLERGENS)
|