From d260853f8a718743146cf9f0f324f67423134e82 Mon Sep 17 00:00:00 2001
From: setop <setop@zoocoop.com>
Date: Mon, 11 Dec 2023 12:36:22 +0100
Subject: [PATCH] day 11, 1 is also 2-1

---
 d11/run.py | 31 +++++++++++++++++++++++++++++++
 1 file changed, 31 insertions(+)
 create mode 100644 d11/run.py

diff --git a/d11/run.py b/d11/run.py
new file mode 100644
index 0000000..a85184b
--- /dev/null
+++ b/d11/run.py
@@ -0,0 +1,31 @@
+import sys
+from itertools import product as Prod
+
+L = [list(l) for l in sys.stdin.read().splitlines()]
+H = len(L)
+W = len(L[0])
+RD = {y for y in range(W) if "".join(L[y]).find("#")==-1}
+CD = {x for x in range(H) if all(L[y][x] == '.' for y in range(W))}
+
+G = [(x,y) 
+for y in range(W)
+    for x in range(H)
+        if L[y][x] == '#']    
+
+X = float(sys.argv[1])  # expantion factor, 2 for part 1, 1E6 for part 2
+
+def exp(a:int,b:int,S:set[int]) -> int:
+    if a > b:
+        b, a = a, b
+    return sum(1 for s in S if a<s<b)
+
+S = 0
+for i in range(len(G)):
+    for j in range(i+1,len(G)):
+        x1,y1 = G[i]
+        x2,y2 = G[j]
+        ex = exp(x1,x2,CD)
+        ey = exp(y1,y2,RD)
+        d = abs(x2-x1)+abs(y2-y1)+(X-1)*(ex+ey)
+        S += d
+print(int(S))