part_two.py

from typing import override

from infrastructure.solutions.base import Solution


class Year2019Day1Part2Solution(Solution):

    @classmethod
    @override
    def parse_input(cls, text_input: str) -> dict[str, list[int]]:
        masses = []

        for line in text_input.split('\n'):
            if line.isdigit():
                masses.append(int(line))

        return {'masses': masses}

    @classmethod
    @override
    def solve(cls, masses: list[int]) -> int:
        """
        Time:  O(n*log3(m))
        Space: O(1)

        Where n - length of list with masses,
              m - maximum mass value
        """
        fuel = 0

        for mass in masses:
            while mass > 0:
                mass = max(0, (mass // 3) - 2)
                fuel += mass

        return fuel


if __name__ == '__main__':
    print(Year2019Day1Part2Solution.main())