From 2f3e4c65766c2cd0e076e3dfc45eed63eb77c2aa Mon Sep 17 00:00:00 2001 From: kageru Date: Sun, 1 Dec 2019 19:37:24 +0100 Subject: [PATCH] Add day 1 in Python --- 2019/1/day1.py | 13 +++++++++++++ 1 file changed, 13 insertions(+) create mode 100644 2019/1/day1.py 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))}')