-
Notifications
You must be signed in to change notification settings - Fork 0
/
7.py
63 lines (54 loc) · 1.7 KB
/
7.py
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
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
bags = []
class Bag:
def __init__(self, color):
self.color = color
self.children = []
bags.append(self)
def add_child(self, amount, bag):
self.children.append((amount, bag))
def __str__(self):
childstr = ""
if self.children:
childrenstr = "[" + ", ".join([str(a) + " of " + str(c) for a, c in self.children]) + "]"
return f"{self.color} -> {childrenstr}"
return f"{self.color}"
def amount_of_children(self):
if not self.children:
return 0
return sum(a + a * c.amount_of_children() for a, c in self.children)
@staticmethod
def find(color):
for bag in bags:
if bag.color == color:
return bag
@staticmethod
def find_or_create(color):
bag = Bag.find(color)
if bag:
return bag
return Bag(color)
def preprocess_data():
with open("input7.txt") as f:
lines = f.readlines()
lines = [x.replace('.', '').replace('bags', '').replace('bag', '') for x in lines]
for l in lines:
node, children = l.split(' contain ')
node = node.strip()
node = Bag.find_or_create(node)
for child in children.split(','):
child = child.strip()
amount, color = child.split(' ', 1)
if amount == "no":
continue
amount = int(amount)
node.add_child(amount, Bag.find_or_create(color.strip()))
preprocess_data()
amount = 0
for bag in bags:
if bag.color == "shiny gold":
continue
if "shiny gold" in str(bag):
amount += 1
print(amount)
shiny = Bag.find("shiny gold")
print(shiny.amount_of_children())