from lib import get_data def f(i): i //= 3 i -= 2 return i def part_1(data): print(sum([f(int(line)) for line in data.splitlines()])) def part_2(data): r = 0 for line in data.splitlines(): i = f(int(line)) while i > 0: r += i i = f(i) print(r) def main(): data = get_data(__file__) part_1(data) part_2(data) if __name__ == "__main__": main()