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)