Add day 1 in Python
This commit is contained in:
parent
7437fcba3a
commit
2f3e4c6576
|
@ -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))}')
|
Loading…
Reference in New Issue
Block a user