blob: c8ca64bbf58bcdaf002791d76efd369b62c6784d (
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
|
#!/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))
|
