Day 7

2020/day7_1.py

@@ -0,0 +1,43 @@
+instr = '''
+A contain 1 C.
+B contain 1 D.
+C contain no other bags.
+D contain 1 E.
+E contain no other bags.
+F contain 1 D.
+'''
+
+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)