From b2c570524c7371e40e21ae107f4b62790607717a Mon Sep 17 00:00:00 2001 From: Aryadev Chavali Date: Sat, 21 Oct 2023 22:34:30 +0100 Subject: Added 2019 puzzles, in python Finished first two rounds of puzzle 1 --- 2019/puzzle-1.py | 24 ++++++++++++++++++++++++ 1 file changed, 24 insertions(+) create mode 100644 2019/puzzle-1.py 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() -- cgit v1.2.3-13-gbd6f