2020 d01 solve

1 files changed, 24 insertions, 0 deletions

diff --git a/20/py/d01.py b/20/py/d01.py
new file mode 100644
index 0000000..c8ca64b
--- /dev/null
+++ b/20/py/d01.py
@@ -0,0 +1,24 @@
+#!/usr/bin/env python3
+def pt1(_in):
+    nums = [int(n) for n in _in]
+    for i, n1 in enumerate(nums):
+        for n2 in nums[i:]:
+            if n1 + n2 == 2020:
+                return n1 * n2
+
+
+def pt2(_in):
+    nums = [int(n) for n in _in]
+    for i, n1 in enumerate(nums):
+        for j, n2 in enumerate(nums[i:]):
+            if n1 + n2 > 2020:
+                continue
+            for n3 in nums[i+j:]:
+                if n1 + n2 + n3 == 2020:
+                    return n1 * n2 * n3
+
+
+if __name__ == "__main__":
+    input = open("../input/01", "r").readlines()
+    print(pt1(input))
+    print(pt2(input))