day 20 part 2

just wanted to get this over with

1 files changed, 23 insertions, 10 deletions

diff --git a/22.20/main.py b/22.20/main.py
index 7cb46de..c6324f5 100644
--- a/22.20/main.py
+++ b/22.20/main.py
@@ -1,16 +1,29 @@
+# brace for horribleness
 from sys import stdin
 a = [int(n) for n in stdin.readlines()]
 
-p = list(range(len(a)))
+def perm(a, p):
+    for i, n in enumerate(a):
+        og_idx = p.index(i)
+        p.pop(og_idx)
+        og_idx = (og_idx + n) % len(p)
+        if og_idx == 0 and n < 0: og_idx = len(p)
+        p.insert(og_idx, i)
+    return p
 
-for i, n in enumerate(a):
-    og_idx = p.index(i)
-    p.pop(og_idx)
-    og_idx = (og_idx + n) % len(p)
-    if og_idx == 0 and n < 0: og_idx = len(p)
-    p.insert(og_idx, i)
+def partOne(a):
+    a = [a[i] for i in perm(a, list(range(len(a))))]
+    def g(n): return a[(a.index(0) + n) % len(a)]
+    return g(1000) + g(2000) + g(3000)
 
-a = [a[i] for i in p]
-def g(n): return a[(a.index(0) + n) % len(a)]
+def partTwo(a):
+    a = [n * 811589153 for n in a]
+    p = list(range(len(a)))
+    for _ in range(10):
+        p = perm(a, p)
+    a = [a[i] for i in p]
+    def g(n): return a[(a.index(0) + n) % len(a)]
+    return g(1000) + g(2000) + g(3000)
 
-print(g(1000) + g(2000) + g(3000))
+print(partOne(a))
+print(partTwo(a))