diff options
| -rw-r--r-- | 20/py/d01.py | 24 |
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)) |