From ad2b66d6540be74194ead0eb67d060312178052d Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Gustav=20S=C3=B6rn=C3=A4s?= Date: Tue, 1 Dec 2020 06:28:46 +0100 Subject: 2020 d01 solve --- 20/py/d01.py | 24 ++++++++++++++++++++++++ 1 file changed, 24 insertions(+) create mode 100644 20/py/d01.py (limited to '20') 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)) -- cgit v1.2.1