34 lines
996 B
Python
34 lines
996 B
Python
input = dict()
|
|
with open('day7_input.txt', 'r') as f:
|
|
for line in f.readlines():
|
|
if len(line.strip()) > 0:
|
|
btype, content = line.strip()[:-1].split(' contain ')
|
|
btype= btype[:-1]
|
|
input[btype] = content.split(', ')
|
|
for i,b in enumerate(input[btype]):
|
|
if b == 'no other bags':
|
|
input[btype] = []
|
|
else:
|
|
num,name = b.split(' ',1)
|
|
if int(num) > 1:
|
|
name = name[0:-1]
|
|
input[btype][i] = (int(num), name)
|
|
|
|
def walk_bags(name, cur, bags):
|
|
if cur == name:
|
|
return True
|
|
elif cur in bags:
|
|
for child in bags[cur]:
|
|
if walk_bags(name, child[1], bags):
|
|
return True
|
|
return False
|
|
else:
|
|
return False
|
|
|
|
|
|
search = 'shiny gold bag'
|
|
count = 0
|
|
for bag in input.keys():
|
|
if bag != search and walk_bags(search, bag, input):
|
|
count += 1
|
|
print(count) |