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