14 lines
322 B
Python
14 lines
322 B
Python
|
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))}')
|