34 lines
835 B
Python
34 lines
835 B
Python
from collections import Counter
|
|
|
|
test_input = """
|
|
3 4
|
|
4 3
|
|
2 5
|
|
1 3
|
|
3 9
|
|
3 3
|
|
""".strip()
|
|
|
|
test_solution_p1 = 11
|
|
test_solution_p2 = 31
|
|
|
|
|
|
def solve_p1(puzzle_input: str) -> int:
|
|
list1, list2 = _parse_lists(puzzle_input)
|
|
pairs = zip(sorted(list1), sorted(list2))
|
|
distances = (abs(pair[0] - pair[1]) for pair in pairs)
|
|
return sum(distances)
|
|
|
|
|
|
def solve_p2(puzzle_input: str) -> int:
|
|
list1, list2 = _parse_lists(puzzle_input)
|
|
occurrences = Counter(list2)
|
|
similarities = (value * occurrences.get(value, 0) for value in list1)
|
|
return sum(similarities)
|
|
|
|
|
|
def _parse_lists(puzzle_input: str) -> tuple[list[int], list[int]]:
|
|
lines = (line.partition(" ") for line in puzzle_input.split("\n") if line)
|
|
list1, list2 = zip(*((int(line[0]), int(line[2])) for line in lines))
|
|
return list1, list2
|