From b65dedcebd761653d8295e4b9356b4eef8e3ca02 Mon Sep 17 00:00:00 2001 From: Nguyễn Gia Phong Date: Wed, 4 Dec 2024 15:22:35 +0900 Subject: [aoc/2024] Finish day 4 --- aoc/2024/04/part-two.py | 10 ++++++++++ 1 file changed, 10 insertions(+) create mode 100644 aoc/2024/04/part-two.py (limited to 'aoc/2024/04/part-two.py') 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))) -- cgit 1.4.1