day7.py

import aoc
from collections import deque

data = aoc.strlist(7)

bags = {}
for i in data:
    outer, inner = i.rstrip(".").split(" contain ")
    outer = outer.rstrip("s")
    bags[outer] = []
    if inner == "no other bags": continue
    for j in inner.split(", "):
        n, col = j.split(" ", 1)
        bags[outer].append((int(n), col.rstrip("s")))

contain = set()
queue = deque(["shiny gold bag"])
while queue:
    i = queue.pop()
    for outer, inner in bags.items():
        for j in inner:
            if j[1] == i:
                contain.add(outer)
                queue.append(outer)

queue = deque([(1, "shiny gold bag")])
count = 0
while queue:
    r, i = queue.pop()
    for j in bags[i]:
        count += r * j[0]
        queue.append((r * j[0], j[1]))

print(len(contain), count)
aoc.tock("ms")