aoc_2024_01.py

(
	solve := lambda data: sum(
		abs(a - b) for a, b in
		zip(*map(sorted,zip(*(
			map(int, line.split())
			for line in data.split("\n")
		))))
	),

	example := """\
		3   4
		4   3
		2   5
		1   3
		3   9
		3   3\
	""",

	challenge := open("input").read().strip(),

	print("part 1:"),
	print("example", solve(example)),
	print("challenge", solve(challenge)),
	print(),

	solve2 := lambda data: (
		(lists := list(zip(*(
			map(int, line.split())
			for line in data.split("\n")
		)))),
		sum(
			n * lists[1].count(n)
			for n in lists[0]
		)
	)[-1],

	print("part 2:"),
	print("example", solve2(example)),
	print("challenge", solve2(challenge)),
)