From b98a41899bec7ae6b992dc11a29285926213b9b0 Mon Sep 17 00:00:00 2001 From: Bruno BELANYI Date: Thu, 19 Dec 2024 00:11:03 -0500 Subject: [PATCH] 2024: d19: ex1: add solution --- 2024/d19/ex1/ex1.py | 32 ++++++++++++++++++++++++++++++++ 1 file changed, 32 insertions(+) create mode 100755 2024/d19/ex1/ex1.py diff --git a/2024/d19/ex1/ex1.py b/2024/d19/ex1/ex1.py new file mode 100755 index 0000000..f8195c5 --- /dev/null +++ b/2024/d19/ex1/ex1.py @@ -0,0 +1,32 @@ +#!/usr/bin/env python + +import functools +import sys + + +def solve(input: str) -> int: + def parse(input: str) -> tuple[frozenset[str], list[str]]: + towels, patterns = input.split("\n\n") + return frozenset(towels.split(", ")), patterns.splitlines() + + @functools.cache + def is_possible(pattern: str, towels: frozenset[str]) -> bool: + if pattern == "": + return True + return any( + is_possible(pattern.removeprefix(towel), towels) + for towel in towels + if pattern.startswith(towel) + ) + + towels, patterns = parse(input) + return sum(is_possible(pattern, towels) for pattern in patterns) + + +def main() -> None: + input = sys.stdin.read() + print(solve(input)) + + +if __name__ == "__main__": + main()