25-02 python p2

1 files changed, 24 insertions, 10 deletions

diff --git a/2025/02/python.py b/2025/02/python.py
index 2e55ec1..120036c 100755
--- a/2025/02/python.py
+++ b/2025/02/python.py
@@ -1,19 +1,33 @@
 #!/usr/bin/env python3
 
-from fileinput import input
-from math import log10
+xs = [
+    str(x)
+    for r in input().split(",")
+    for [a, b] in [r.split("-")]
+    for x in range(int(a), int(b) + 1)
+]
 
-rs = [[int(i) for i in r.split("-")] for r in "".join(input()).split(",")]
 
+def invalid(x: str, n: int):
+    p, q = divmod(len(x), n)
+    return (
+        p != 0
+        and q == 0
+        and len(
+            {
+                x[a:b]
+                for a, b in zip(
+                    range(0, len(x) + 1, p),
+                    range(p, len(x) + 1, p),
+                )
+            }
+        )
+        == 1
+    )
 
-def invalid(i):
-    d = int(log10(i)) + 1
-    a, b = divmod(i, 10 ** (d // 2))
-    return a == b
 
-
-silver = sum(x for [a, b] in rs for x in range(a, b + 1) if invalid(x))
-gold = 0
+silver = sum(int(x) for x in xs if invalid(x, 2))
+gold = sum(int(x) for x in xs if any(invalid(x, n) for n in range(2, len(x) + 1)))
 
 print("silver:", silver)
 print("gold:", gold)