Kaynağa Gözat

Simplify day 11

Taddeus Kroes 1 yıl önce
ebeveyn
işleme
5aad4b3c75
1 değiştirilmiş dosya ile 6 ekleme ve 6 silme
  1. 6 6
      2024/11_pebbles.py

+ 6 - 6
2024/11_pebbles.py

@@ -6,13 +6,13 @@ def change(stone):
     if stone == 0:
         yield 1
     else:
-        s = str(stone)
-        l = len(s)
-        if l % 2 == 0:
-            yield int(s[:l // 2])
-            yield int(s[l // 2:])
-        else:
+        digits = str(stone)
+        mid, odd = divmod(len(digits), 2)
+        if odd:
             yield stone * 2024
+        else:
+            yield int(digits[:mid])
+            yield int(digits[mid:])
 
 def blink(counts, times):
     for _ in range(times):