diff options
Diffstat (limited to '2019/puzzle-1.py')
-rw-r--r-- | 2019/puzzle-1.py | 24 |
1 files changed, 24 insertions, 0 deletions
diff --git a/2019/puzzle-1.py b/2019/puzzle-1.py new file mode 100644 index 0000000..a6fffbd --- /dev/null +++ b/2019/puzzle-1.py @@ -0,0 +1,24 @@ +from math import floor + +def fuel(mass): + return (floor(mass / 3)) - 2 + +def read_input(): + with open("1-input", "r") as fp: + return [int(line) for line in fp.readlines()] + +def round_1(): + print(f"round 1: {sum([fuel(mass) for mass in read_input()])}") + +def recursive_fuel(mass): + new_fuel = (floor(mass / 3)) - 2 + if new_fuel <= 0: + return 0 + return new_fuel + recursive_fuel(new_fuel) + +def round_2(): + print(f"round 2: {sum([recursive_fuel(mass) for mass in read_input()])}") + +if __name__ == '__main__': + round_1() + round_2() |