Faster day 4

1 files changed, 19 insertions, 7 deletions

diff --git a/solutions/py/d04.py b/solutions/py/d04.py
index e3fb41a..778b05e 100644
--- a/solutions/py/d04.py
+++ b/solutions/py/d04.py
@@ -1,13 +1,17 @@
 from collections import Counter
 
 def isIncreasing(num):
-    s = str(num)
+    s = list(str(num))
     n = int(s[0])
+    n_i = 0
     for sp in s[1:]:
+        n_i += 1
         if int(sp) < n:
-            return False
+            for i in range(n_i, 6):
+                s[i] = str(n)
+            return False, int("".join(s))
         n = int(sp)
-    return True
+    return (True,)
 
 def pt1(input):
     def containsDouble(num):
@@ -19,10 +23,14 @@ def pt1(input):
         return c[0] + c[1] < 10
 
     amount = 0
-    for n in range(357253, 892942 + 1):
-        if isIncreasing(n):
+    n = 357253
+    while n < 892942 + 1:
+        if (inc := isIncreasing(n))[0] == True:
             if containsDouble(n):
                 amount += 1
+            n += 1
+        else:
+            n = inc[1]
     return amount
 
 def pt2(input):
@@ -36,10 +44,14 @@ def pt2(input):
             return c[2] >= 1
 
     amount = 0
-    for n in range(357253, 892942 + 1):
-        if isIncreasing(n):
+    n = 357253
+    while n < 892942 + 1:
+        if (inc := isIncreasing(n))[0] == True:
             if containsDouble(n):
                 amount += 1
+            n += 1
+        else:
+            n = inc[1]
     return amount
 
 if __name__ == "__main__":