aoc2019/day01.py
2021-12-27 17:31:33 +01:00

42 lines
924 B
Python

from tools.aoc import AOCDay
class Day(AOCDay):
inputs = [
[
(2, "test_input01"),
(2, "test_input01_2"),
(654, "test_input01_3"),
(33583, "test_input01_4"),
(3282386, "input01")
],
[
(2, "test_input01_2"),
(966, "test_input01_3"),
(50346, "test_input01_4"),
(4920708, "input01")
]
]
def part1(self):
fuel_sum = 0
for x in self.getInputListAsType(int):
fuel_sum += x // 3 - 2
return fuel_sum
def part2(self):
fuel_sum = 0
for x in self.getInputListAsType(int):
fuel_add = x // 3 - 2
while fuel_add > 0:
fuel_sum += fuel_add
fuel_add = fuel_add // 3 - 2
return fuel_sum
if __name__ == '__main__':
day = Day(2019, 1)
day.run(verbose=True)