From ae8759d766439c7186fbe6d0c73d8a9c793ed1e7 Mon Sep 17 00:00:00 2001 From: Bruno BELANYI Date: Sun, 1 Dec 2024 08:48:02 +0000 Subject: [PATCH] 2024: d01: ex2: add solution --- 2024/d01/ex2/ex2.py | 25 +++++++++++++++++++++++++ 1 file changed, 25 insertions(+) create mode 100755 2024/d01/ex2/ex2.py diff --git a/2024/d01/ex2/ex2.py b/2024/d01/ex2/ex2.py new file mode 100755 index 0000000..2f51f10 --- /dev/null +++ b/2024/d01/ex2/ex2.py @@ -0,0 +1,25 @@ +#!/usr/bin/env python + +import sys + + +def solve(input: list[str]) -> int: + def parse(input: list[str]) -> tuple[list[int], list[int]]: + left, right = [], [] + for line in input: + lhs, rhs = line.split() + left.append(int(lhs)) + right.append(int(rhs)) + return left, right + + left, right = parse(input) + return sum(lhs * right.count(lhs) for lhs in left) + + +def main() -> None: + input = sys.stdin.read().splitlines() + print(solve(input)) + + +if __name__ == "__main__": + main()