54 lines
1.6 KiB
Python
54 lines
1.6 KiB
Python
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()
|