Added 2019 puzzles, in python
Finished first two rounds of puzzle 1
This commit is contained in:
24
2019/puzzle-1.py
Normal file
24
2019/puzzle-1.py
Normal file
@@ -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()
|
||||
Reference in New Issue
Block a user