Add day 1 in Python

This commit is contained in:
kageru 2019-12-01 19:37:24 +01:00
parent 7437fcba3a
commit 2f3e4c6576
Signed by: kageru
GPG Key ID: 8282A2BEA4ADA3D2

13
2019/1/day1.py Normal file
View File

@ -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))}')