22-1

koosvary · Dec 22, 2024 · 76656e7 · 76656e7
1 parent ed2e4f1
commit 76656e7
Show file tree

Hide file tree

Showing 2 changed files with 35 additions and 0 deletions.
diff --git a/2024/22/22-1.py b/2024/22/22-1.py
@@ -0,0 +1,31 @@
+import math
+
+def mixThenPrune(secretNumber, multiplying, mutator):
+    if multiplying:
+        newSecretNumber = secretNumber * mutator
+    else:
+        newSecretNumber = math.floor(secretNumber / mutator)
+
+    # mix
+    newSecretNumber = secretNumber ^ newSecretNumber
+
+    # prune
+    newSecretNumber %= 16777216
+
+    return newSecretNumber
+
+total = 0
+with open('2024/22/input.txt') as f:
+    for line in f:
+        line = line.strip()
+
+        secretNumber = int(line)
+        for i in range(2000):
+            secretNumber = mixThenPrune(secretNumber, True, 64)
+            secretNumber = mixThenPrune(secretNumber, False, 32)
+            secretNumber = mixThenPrune(secretNumber, True, 2048)
+
+        total += secretNumber
+
+print(total)
+
diff --git a/2024/22/testinput.txt b/2024/22/testinput.txt
@@ -0,0 +1,4 @@
+1
+10
+100
+2024