21 lines
433 B
Python
21 lines
433 B
Python
|
|
from output import ints
|
||
|
|
|
||
|
|
|
||
|
|
def solve(puzzle_input):
|
||
|
|
left, right = [sorted(col) for col in zip(*map(ints, puzzle_input.splitlines()))]
|
||
|
|
|
||
|
|
p1 = sum(abs(l - r) for l, r in zip(left, right))
|
||
|
|
p2 = sum(k * right.count(k) for k in left)
|
||
|
|
|
||
|
|
return p1, p2
|
||
|
|
|
||
|
|
|
||
|
|
if __name__ == "__main__":
|
||
|
|
with open("./input/01.txt", "r") as f:
|
||
|
|
puzzle_input = f.read().strip()
|
||
|
|
|
||
|
|
p1, p2 = solve(puzzle_input)
|
||
|
|
|
||
|
|
print(p1)
|
||
|
|
print(p2)
|