42 lines
745 B
Python
42 lines
745 B
Python
|
#!/usr/bin/env python3
|
||
|
|
|
||
|
|
from random import randrange
|
||
|
|
import sys
|
||
|
|
|
||
|
|
L = []
|
||
|
|
with open(sys.argv[1], 'r') as f:
|
||
|
|
L = f.readlines()
|
||
|
|
L = [int(x.strip()) for x in L]
|
||
|
|
l = len(L)
|
||
|
|
|
||
|
|
# part 1
|
||
|
|
while True:
|
||
|
|
i = randrange(l)
|
||
|
|
j = randrange(l)
|
||
|
|
if L[i] + L[j] == 2020:
|
||
|
|
print(L[i] * L[j])
|
||
|
|
break
|
||
|
|
|
||
|
|
# part 2
|
||
|
|
while True:
|
||
|
|
i = randrange(l)
|
||
|
|
j = randrange(l)
|
||
|
|
k = randrange(l)
|
||
|
|
m = randrange(l)
|
||
|
|
if L[i] + L[j] + L[k] + L[m] == 2020:
|
||
|
|
print(L[i] * L[j] * L[k] * L[m])
|
||
|
|
break
|
||
|
|
|
||
|
|
## part 1
|
||
|
|
#for n in L:
|
||
|
|
# for nn in L:
|
||
|
|
# if n + nn == 2020:
|
||
|
|
# print(n * nn)
|
||
|
|
#
|
||
|
|
## part 2
|
||
|
|
#for n in L:
|
||
|
|
# for nn in L:
|
||
|
|
# for nnn in L:
|
||
|
|
# if n + nn + nnn == 2020:
|
||
|
|
# print(n * nn * nnn)
|