Update day01.py
This commit is contained in:
parent
20236579d9
commit
d6935aa828
|
@ -1,12 +1,24 @@
|
||||||
def solve_p2():
|
def solve_p1():
|
||||||
numbers = []
|
numbers = []
|
||||||
with open("input.txt") as f:
|
with open("input.txt") as f:
|
||||||
for line in f:
|
for line in f:
|
||||||
num = int(line)
|
num = int(line)
|
||||||
|
opp = 2020-num
|
||||||
|
if opp in numbers:
|
||||||
|
print(num*opp)
|
||||||
numbers.append(num)
|
numbers.append(num)
|
||||||
|
|
||||||
|
|
||||||
|
def solve_p2():
|
||||||
|
numbers = []
|
||||||
|
with open("input.txt") as f:
|
||||||
|
for line in f:
|
||||||
|
numbers.append(int(line))
|
||||||
combos = [((a, b, c), a*b*c) if a+b+c == 2020 else None for (a, b, c) in itertools.combinations(numbers, 3)]
|
combos = [((a, b, c), a*b*c) if a+b+c == 2020 else None for (a, b, c) in itertools.combinations(numbers, 3)]
|
||||||
for combo in combos:
|
for combo in combos:
|
||||||
if combo is not None:
|
if combo is not None:
|
||||||
print(combo)
|
print(combo)
|
||||||
|
|
||||||
|
|
||||||
|
solve_p1():
|
||||||
solve_p2()
|
solve_p2()
|
||||||
|
|
Loading…
Reference in a new issue