aoc2019/day01.py
2021-11-26 08:10:55 +01:00

24 lines
540 B
Python

from aoc import AOCDay
class Day(AOCDay):
test_solutions_p1 = [2, 2, 654, 33583]
test_solutions_p2 = [2, 966, 50346]
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