summaryrefslogtreecommitdiff
diff options
context:
space:
mode:
-rw-r--r--2024/11/python/main.py28
1 files changed, 28 insertions, 0 deletions
diff --git a/2024/11/python/main.py b/2024/11/python/main.py
new file mode 100644
index 0000000..445809a
--- /dev/null
+++ b/2024/11/python/main.py
@@ -0,0 +1,28 @@
+from fileinput import input
+from functools import cache
+from math import floor, log
+
+stones = [int(x) for x in input().readline().split()]
+
+
+@cache
+def blink(s, n):
+ if n == 0:
+ return 1
+
+ if s == 0:
+ return blink(1, n - 1)
+
+ digits = floor(log(s, 10) + 1e-6) + 1
+ if digits % 2 == 0:
+ left, right = divmod(s, 10 ** (digits // 2))
+ return blink(left, n - 1) + blink(right, n - 1)
+
+ return blink(s * 2024, n - 1)
+
+
+silver = sum(blink(s, 25) for s in stones)
+gold = sum(blink(s, 75) for s in stones)
+
+print("silver:", silver)
+print("gold:", gold)