1 files changed, 28 insertions, 0 deletions
diff --git a/19/py/d01.py b/19/py/d01.py
new file mode 100644
index 0000000..d1c57c8
--- /dev/null
+++ b/19/py/d01.py
@@ -0,0 +1,28 @@
+import math
+import sys
+
+def get_fuel(mass):
+    fuel = math.floor(mass / 3) - 2
+    return 0 if fuel <= 0 else fuel + get_fuel(fuel)
+
+def pt1(_in):
+    s = 0
+    for line in _in:
+        mass = int(line)
+        if mass == 0:
+            break
+        fuel = math.floor(mass / 3) - 2
+        s += fuel
+    return s
+
+def pt2(input):
+    return sum([get_fuel(int(line)) for line in input])
+
+if __name__ == "__main__":
+    import cProfile
+
+    input = open("../input/01", "r").readlines()
+    cProfile.run("pt1(input)")
+    cProfile.run("pt2(input)")
+    print(pt1(input))
+    print(pt2(input))