advent/2017/day02.py

27 lines
453 B
Python
Raw Permalink Normal View History

2017-12-14 00:48:23 +00:00
with open('data02.txt') as f:
rows = f.readlines()
'''
# part 1
diffs = []
for r in rows:
cells = [int(c) for c in r.split()]
diffs.append(max(cells) - min(cells))
print(sum(diffs))
'''
# part 2
from itertools import combinations
rows = [[int(x) for x in r.split()] for r in rows]
divs = []
for row in rows:
for pair in combinations(row, 2):
hi, lo = max(pair), min(pair)
if hi % lo == 0:
divs.append(hi // lo)
break
print(sum(divs))