[aoc/2024] Finish day 4

1 files changed, 10 insertions, 0 deletions

diff --git a/aoc/2024/04/part-two.py b/aoc/2024/04/part-two.py
new file mode 100644
index 0000000..a3e7c4a
--- /dev/null
+++ b/aoc/2024/04/part-two.py
@@ -0,0 +1,10 @@
+from itertools import islice
+from sys import stdin
+
+mat = tuple(map(str.strip, stdin.readlines()))
+m = len(mat)
+n = len(mat[0])
+print(sum(c == 'A' and ({mat[i-1][j-1], mat[i+1][j+1]}
+                        == {mat[i-1][j+1], mat[i+1][j-1]} == {'M', 'S'})
+          for i, line in islice(enumerate(mat), 1, m-1)
+          for j, c in islice(enumerate(line), 1, n-1)))