blob: 932f370d3ccc35a8465e70a3ca6ebbfc70d7c750 (
plain) (
blame)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
|
#!/usr/bin/env python3
import aoc20
import sys
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 = aoc20.read_input(sys.argv[1:], 1)
print(pt1(input))
print(pt2(input))
|
