From 15e50e8dec52b35628a0e26ffa7e63e630fce2a7 Mon Sep 17 00:00:00 2001
From: Bruno BELANYI <bruno@belanyi.fr>
Date: Wed, 4 Dec 2024 10:36:07 +0000
Subject: [PATCH] 2024: d04: ex2: add solution

---
 2024/d04/ex2/ex2.py | 49 +++++++++++++++++++++++++++++++++++++++++++++
 1 file changed, 49 insertions(+)
 create mode 100755 2024/d04/ex2/ex2.py

diff --git a/2024/d04/ex2/ex2.py b/2024/d04/ex2/ex2.py
new file mode 100755
index 0000000..fc9b4f3
--- /dev/null
+++ b/2024/d04/ex2/ex2.py
@@ -0,0 +1,49 @@
+#!/usr/bin/env python
+
+import itertools
+import sys
+from typing import NamedTuple
+
+
+class Point(NamedTuple):
+    x: int
+    y: int
+
+
+def solve(input: str) -> int:
+    def parse(input: list[str]) -> dict[Point, str]:
+        return {
+            Point(x, y): c for x, line in enumerate(input) for y, c in enumerate(line)
+        }
+
+    def count_xmas(grid: dict[Point, str], dims: Point) -> int:
+        max_x, max_y = dims
+        count = 0
+        for x, y in itertools.product(range(max_x), range(max_y)):
+            if grid.get(Point(x, y)) != "A":
+                continue
+            if {grid.get(Point(x + dx, y + dy)) for dx, dy in ((-1, -1), (1, 1))} != {
+                "M",
+                "S",
+            }:
+                continue
+            if {grid.get(Point(x + dx, y + dy)) for dx, dy in ((1, -1), (-1, 1))} != {
+                "M",
+                "S",
+            }:
+                continue
+            count += 1
+        return count
+
+    lines = input.splitlines()
+    grid = parse(lines)
+    return count_xmas(grid, Point(len(lines), len(lines[0])))
+
+
+def main() -> None:
+    input = sys.stdin.read()
+    print(solve(input))
+
+
+if __name__ == "__main__":
+    main()