From 4e46f615cf078f2204f607f451a880a1d7cfc75c Mon Sep 17 00:00:00 2001
From: Bruno BELANYI <bruno@belanyi.fr>
Date: Sun, 12 Dec 2021 15:13:10 +0100
Subject: [PATCH] 2021: d12: ex1: add solution

---
 2021/d12/ex1/ex1.py | 46 +++++++++++++++++++++++++++++++++++++++++++++
 1 file changed, 46 insertions(+)
 create mode 100755 2021/d12/ex1/ex1.py

diff --git a/2021/d12/ex1/ex1.py b/2021/d12/ex1/ex1.py
new file mode 100755
index 0000000..4d0cacf
--- /dev/null
+++ b/2021/d12/ex1/ex1.py
@@ -0,0 +1,46 @@
+#!/usr/bin/env python
+
+import itertools
+import sys
+from collections import defaultdict
+from typing import Dict, List, Set
+
+Map = Dict[str, Set[str]]
+
+
+def solve(input: List[str]) -> int:
+    def parse() -> Map:
+        res: Map = defaultdict(set)
+
+        for (start, to) in map(lambda s: s.split("-"), input):
+            res[start].add(to)
+            res[to].add(start)
+
+        return res
+
+    caves = parse()
+
+    def dfs(start: str, seen: Set[str] = set()) -> int:
+        if start == "end":
+            return 1
+
+        seen = seen | {start}
+        res = 0
+
+        for dest in caves[start]:
+            if dest in seen and dest.islower():
+                continue
+            res += dfs(dest, seen)
+
+        return res
+
+    return dfs("start")
+
+
+def main() -> None:
+    input = [line.strip() for line in sys.stdin.readlines()]
+    print(solve(input))
+
+
+if __name__ == "__main__":
+    main()