Day 7

Author: Programator2

2020/7.txt

@@ -0,0 +1,9 @@
+light red bags contain 1 bright white bag, 2 muted yellow bags.
+dark orange bags contain 3 bright white bags, 4 muted yellow bags.
+bright white bags contain 1 shiny gold bag.
+muted yellow bags contain 2 shiny gold bags, 9 faded blue bags.
+shiny gold bags contain 1 dark olive bag, 2 vibrant plum bags.
+dark olive bags contain 3 faded blue bags, 4 dotted black bags.
+vibrant plum bags contain 5 faded blue bags, 6 dotted black bags.
+faded blue bags contain no other bags.
+dotted black bags contain no other bags.

2020/7_test.txt

@@ -0,0 +1,7 @@
+shiny gold bags contain 2 dark red bags.
+dark red bags contain 2 dark orange bags.
+dark orange bags contain 2 dark yellow bags.
+dark yellow bags contain 2 dark green bags.
+dark green bags contain 2 dark blue bags.
+dark blue bags contain 2 dark violet bags.
+dark violet bags contain no other bags.

2020/7_test2.txt

@@ -0,0 +1,57 @@
+from aocd import submit
+import re
+from aoc import *
+from collections import defaultdict
+
+
+def submita(answer):
+    submit(answer, part='a', day=7, year=2020)
+
+
+def submitb(answer):
+    submit(answer, part='b', day=7, year=2020)
+
+
+FILE = '7_test.txt'
+FILE = '7.txt'
+
+
+class Bag:
+    def __init__(self):
+        self.name = None
+        self.parents = set()
+        self.children = {}
+
+
+def backtrack(bag, s, bags, depth):
+    for p in bag.parents:
+        if p not in s:
+            s.add(p)
+            backtrack(bags[p], s, bags, depth+1)
+
+
+def main():
+    inp = lines(FILE)
+    bags = defaultdict(Bag)
+    for l in inp:
+        b = re.match(r'(.+ .+) bags contain (.+)\.|(.+ .+) bags contain no other bags\.', l)
+        outer_bag = b[1]
+        inside = b[2]
+        if inside != 'no other bags':
+            ins = re.findall(r'(\d+) (.+? .+?) bags?(?:, )?', inside)
+            for number, name in ins:
+                bags[outer_bag].children[name] = int(number)
+                bags[name].parents.add(outer_bag)
+    for k, v in bags.items():
+        v.name = k
+    print('OK')
+    contain_gold = set()
+    backtrack(bags['shiny gold'], contain_gold, bags, 1)
+    out = len(contain_gold)
+    print(out)
+    input()
+    print('submitting')
+    submita(out)
+
+
+main()

2020/7a.py

@@ -0,0 +1,58 @@
+from aocd import submit
+import re
+from aoc import *
+from collections import defaultdict
+
+
+def submita(answer):
+    submit(answer, part='a', day=7, year=2020)
+
+
+def submitb(answer):
+    submit(answer, part='b', day=7, year=2020)
+
+
+FILE = '7_test.txt'
+FILE = '7_test2.txt'
+FILE = '7.txt'
+
+
+class Bag:
+    def __init__(self):
+        self.name = None
+        self.parents = set()
+        self.children = {}
+
+
+def downtrack(bag, bags):
+    number = 1
+    for p, n in bag.children.items():
+        print(p, n)
+        number += n * downtrack(bags[p], bags)
+    return number
+
+
+def main():
+    inp = lines(FILE)
+    bags = defaultdict(Bag)
+    for l in inp:
+        b = re.match(r'(.+ .+) bags contain (.+)\.|(.+ .+) bags contain no other bags\.', l)
+        outer_bag = b[1]
+        inside = b[2]
+        if inside != 'no other bags':
+            ins = re.findall(r'(\d+) (.+? .+?) bags?(?:, )?', inside)
+            for number, name in ins:
+                bags[outer_bag].children[name] = int(number)
+                bags[name].parents.add(outer_bag)
+    for k, v in bags.items():
+        v.name = k
+    print('OK')
+    out = downtrack(bags['shiny gold'], bags)
+    out -= 1
+    print(out)
+    input()
+    print('submitting')
+    submitb(out)
+
+
+main()