From 1f357ee9a0e3204512c1e701cba1d4470bb4abad Mon Sep 17 00:00:00 2001 From: Bruno BELANYI Date: Wed, 3 Dec 2025 08:19:08 +0000 Subject: [PATCH] 2025: d03: ex2: add solution --- 2025/d03/ex2/ex2.py | 32 ++++++++++++++++++++++++++++++++ 1 file changed, 32 insertions(+) create mode 100755 2025/d03/ex2/ex2.py diff --git a/2025/d03/ex2/ex2.py b/2025/d03/ex2/ex2.py new file mode 100755 index 0000000..d8358fc --- /dev/null +++ b/2025/d03/ex2/ex2.py @@ -0,0 +1,32 @@ +#!/usr/bin/env python + +import sys + + +def solve(input: list[str]) -> int: + def parse(input: list[str]) -> list[list[int]]: + return [[int(c) for c in line] for line in input] + + def max_joltage(bank: list[int]) -> int: + def recurse(bank: list[int], length: int) -> int: + assert length > 0 # Sanity check + assert length <= len(bank) # Sanity check + if length == 1: + return max(bank) + first = bank.index(max(bank[: -length + 1])) + rest = bank[first + 1 :] + return bank[first] * 10 ** (length - 1) + recurse(rest, length - 1) + + return recurse(bank, 12) + + batteries = parse(input) + return sum(map(max_joltage, batteries)) + + +def main() -> None: + input = sys.stdin.read().splitlines() + print(solve(input)) + + +if __name__ == "__main__": + main()