diff --git a/2019/1/day1.py b/2019/1/day1.py new file mode 100644 index 0000000..34eda6b --- /dev/null +++ b/2019/1/day1.py @@ -0,0 +1,13 @@ +import sys +from typing import List + +def cost(mass: int) -> int: + return mass // 3 - 2 + +def cost_rec(mass: int) -> int: + c = cost(mass) + return 0 if c <= 0 else c + cost_rec(c) + +lines = [int(l) for l in sys.stdin.readlines()] +print(f'Part 1: {sum(map(cost, lines))}') +print(f'Part 2: {sum(map(cost_rec, lines))}')