blob: 1f265377747821bdbed059213e44b9de3bef2cad (
plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
|
"""Day 7."""
import re
from collections import defaultdict
pattern = r"(\d+) ([a-z]+) ([a-z]+) (bag[s]?)"
prog = re.compile(pattern)
with open("input") as f:
sections = f.read().rstrip().split("\n")
mapping = defaultdict(lambda: [])
# Put bags into dict structure
for section in sections:
parent, child_str = section.split(" bags contain ")
children = []
if "no other bags" not in child_str:
for child in child_str.replace(".", "").split(", "):
_, adjective, color, _ = prog.match(child).groups()
children.append(f"{adjective} {color}")
for child in children:
mapping[child].append(parent)
# Part 1
bags = set(mapping['shiny gold'])
while True:
new_bags = bags.copy()
for bag in bags:
for new_bag in mapping[bag]:
new_bags.add(new_bag)
if len(new_bags) == len(bags):
print(len(bags))
break
else:
bags = new_bags
|