Day 14

2019/day14.py

@@ -0,0 +1,53 @@
+import math
+import re
+
+
+def main():
+    with open("day14.txt", "rt") as input_file:
+        lines = [line.rstrip() for line in input_file.readlines()]
+    line_re = re.compile(r"(.+) => (.+)")
+    raw_transmos = [line_re.match(line).groups() for line in lines]
+    transmos = {}
+    for inputs, outputs in raw_transmos:
+        mult, result = outputs.split(" ")
+        input_comps = [i.split(" ") for i in inputs.split(", ")]
+        input_tuples = [(int(q), n) for q, n in input_comps]
+        transmos[result] = int(mult), input_tuples
+
+    print("Required ore for 1 FUEL:", get_required_ore(transmos, "FUEL", 1, {}))
+    
+def get_required_ore(transmos, name, quantity, rabs):
+    if name == "ORE":
+        return quantity
+
+    print("What is required for {} {}?".format(quantity, name))
+    mult, comps = transmos[name]
+    num_reactions = math.ceil(quantity / mult)
+
+    created = 0
+    if name in rabs:
+        print("  Found {} in rab.".format(rabs[name]))
+        created += rabs[name]
+        rabs[name] = 0
+
+    req_ore = 0
+    for _ in range(num_reactions):
+        if created >= quantity:
+            break
+        print("  Creation of {} {}...".format(mult, name))
+        for rq, rn in comps:
+            req_ore += get_required_ore(transmos, rn, rq, rabs)
+        created += mult
+    print("  Possessing {} {}.".format(created, name))
+
+    if created > quantity:
+        leftover = created - quantity
+        rabs[name] = rabs.get(name, 0) + leftover
+        print("  Stored {} {} in rab.".format(leftover, name))
+    
+    print("Answer ({} {}): {} ores".format(quantity, name, req_ore))
+    return req_ore
+
+
+if __name__ == "__main__":
+    main()