diff --git a/2024-python/output/day_01.py b/2024-python/output/day_01.py new file mode 100644 index 0000000..104aba3 --- /dev/null +++ b/2024-python/output/day_01.py @@ -0,0 +1,20 @@ +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)