From a5b0fe494dfea1d96c3ca23731ff3f467d97fefc Mon Sep 17 00:00:00 2001
From: Fedor Lyanguzov <fedor.lyanguzov@yandex.ru>
Date: Sun, 8 Dec 2024 08:30:58 +0300
Subject: [PATCH] 08.1, 08.2 python

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

diff --git a/08/08.py b/08/08.py
new file mode 100644
index 0000000..60cd581
--- /dev/null
+++ b/08/08.py
@@ -0,0 +1,46 @@
+from collections import defaultdict
+import string
+from itertools import combinations
+
+data = open("input.txt").read().strip().split("\n")
+
+d = defaultdict(list)
+t = set(string.ascii_letters+string.digits)
+for i, l in enumerate(data):
+    for j, c in enumerate(l):
+        if c in t:
+            d[c].append((i, j))
+width = len(data[0])
+height = len(data)
+
+def bound(x, width, height):
+    i, j = x
+    return 0<=i<height and 0<=j<width
+
+def f(d):
+    for v in d.values():
+        for (ia, ja), (ib, jb) in combinations(v, 2):
+            di = ib-ia
+            dj = jb-ja
+            yield (ib+di, jb+dj)
+            yield (ia-di, ja-dj)
+
+print(len(set(filter(lambda x: bound(x, width, height), f(d)))))
+
+
+def f2(d):
+    for v in d.values():
+        for (ia, ja), (ib, jb) in combinations(v, 2):
+            di = ib-ia
+            dj = jb-ja
+            x = 0
+            while bound((ib+x*di, jb+x*dj), width, height):
+                yield (ib+x*di, jb+x*dj)
+                x += 1
+            x = 0
+            while bound((ia-x*di, ja-x*dj), width, height):
+                yield (ia-x*di, ja-x*dj)
+                x += 1
+
+
+print(len(set(f2(d))))