Files
aocpy/2020/d1.py

32 lines
704 B
Python

from lib import get_data, str_to_ints
def part_1(data):
xs = sorted(str_to_ints(data))
for i in range(len(xs)):
for j in range(i + 1, len(xs)):
a, b = xs[i], xs[j]
if a + b == 2020:
print(a * b)
if a + b > 2020:
break
for i in range(len(xs)):
for j in range(i + 1, len(xs)):
for k in range(j + 1, len(xs)):
a, b, c = xs[i], xs[j], xs[k]
if a + b + c == 2020:
print(a * b * c)
if a + b + c > 2020:
break
def main():
data = get_data(__file__)
part_1(data)
if __name__ == "__main__":
main()