diff options
Diffstat (limited to '22.20/main.py')
-rw-r--r-- | 22.20/main.py | 16 |
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)) |