day 20 part 1

1 files changed, 16 insertions, 0 deletions

diff --git a/22.20/main.py b/22.20/main.py
new file mode 100644
index 0000000..7cb46de
--- /dev/null
+++ b/22.20/main.py
@@ -0,0 +1,16 @@
+from sys import stdin
+a = [int(n) for n in stdin.readlines()]
+
+p = list(range(len(a)))
+
+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)
+
+a = [a[i] for i in p]
+def g(n): return a[(a.index(0) + n) % len(a)]
+
+print(g(1000) + g(2000) + g(3000))