advent-of-code/2024-python/output/day_01.py

21 lines
433 B
Python
Raw Normal View History

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)