From 7751368d021d9ee7a6fc729478fb84604ea03115 Mon Sep 17 00:00:00 2001
From: Fedor Lyanguzov <Fedor-Lyanguzov@users.noreply.github.com>
Date: Tue, 10 Dec 2024 14:31:24 +0300
Subject: [PATCH] 10.1, 10.2 python

---
 10/10.py | 42 ++++++++++++++++++++++++++++++++++++++++++
 1 file changed, 42 insertions(+)
 create mode 100644 10/10.py

diff --git a/10/10.py b/10/10.py
new file mode 100644
index 0000000..5ce75c0
--- /dev/null
+++ b/10/10.py
@@ -0,0 +1,42 @@
+
+
+
+data = open("input.txt").read().strip().split("\n")
+data = [list(map(int, x)) for x in data]
+
+def make_bound(data):
+    N = len(data)
+    M = len(data)
+    def bound(i, j):
+        return 0<=i<N and 0<=j<M
+    return bound
+
+def f1(data, collect=set):
+    def score(i, j, x=0):
+        if not bound(i, j):
+            return
+        if data[i][j]!=x:
+            return
+        if x==9:
+            yield (i, j)
+        for di, dj in d:
+            yield from score(i+di, j+dj, x+1)
+    zeroes = [(i, j) for i, l in enumerate(data) for j, x in enumerate(l) if x==0]
+    bound = make_bound(data)
+    d = [(1,0), (0,1), (-1,0), (0,-1)]
+    return sum(1 for z in zeroes for _ in collect(score(*z)))
+print(f1(data))
+d = '''
+89010123
+78121874
+87430965
+96549874
+45678903
+32019012
+01329801
+10456732
+'''.strip().split("\n")
+d = [list(map(int, x)) for x in d]
+print(f1(d))
+
+print(f1(data, list))