Day 7 (crunchncrisp.py)

day07/crunchncrisp.py

@@ -0,0 +1,50 @@
+import collections
+import datetime
+
+if __name__ == '__main__':
+
+
+    with open('input.txt') as file:
+        bag_list = [x for x in file.read().splitlines()]
+
+        color_contains = collections.defaultdict(list)
+        color_contained_in = collections.defaultdict(set)
+
+        for bag in bag_list:
+            bag_color  = bag.split(" bags contain ")[0]
+            bag_content = bag.split(" bags contain ")[1].split(", ")
+
+            for content in bag_content:
+                amount = content.split(" ", 1)[0]
+                if amount == "no":
+                    amount = 0
+                else:
+                    amount = int(amount)
+                color = content.split(" ", 1)[1].split(" bag")[0]
+
+                color_contains[bag_color].append((amount, color))
+                color_contained_in[color].add(bag_color)
+
+
+        # Part One
+        def check_for_colors(color):
+            gold_bags = set()
+            for bag_color in color_contained_in[color]:
+                gold_bags.add(bag_color)
+                gold_bags |= check_for_colors(bag_color)
+            return gold_bags
+        print(len(check_for_colors("shiny gold")))
+
+        # Part Two
+        def check_for_amount(color, color_amount):
+            total = 0
+            for a, c in color_contains[color]:
+                total += a
+                if c in color_amount:
+                    total += a * color_amount[c]
+                else:
+                    total += a * check_for_amount(c, color_amount)
+            color_amount[color] = total
+            return total
+
+    print(check_for_amount("shiny gold", {}))