from lib import *

EXAMPLE = """1
2
-3
3
-2
0
4
"""


def move(xs, i, move):
    move = move % (len(xs) - 1)
    ni = (i + move + 1) % len(xs)
    xsi, xs[i] = xs[i], None
    xs.insert(ni, xsi)
    xs.remove(None)
    return xs


def asserts():
    assert move([1, 2, 3], 0, 0) == [1, 2, 3]
    assert move([1, 2, 3], 0, 1) == [2, 1, 3]
    assert move([1, 2, 3], 0, 2) == [1, 2, 3]
    assert move([1, 2, 3], 0, 3) == [2, 1, 3]
    assert move([1, 2, 3], 0, 4) == [1, 2, 3]
    assert move([1, 2, 3], 0, 5) == [2, 1, 3]
    assert move([1, 2, 3], 0, 2) == [1, 2, 3]
    assert move([3, 2, 1], 2, 1) == [3, 1, 2]
    assert move([3, 2, 1], 2, 2) == [1, 3, 2]

    assert move([4, -2, 5, 6, 7, 8, 9,], 1, -2) == [4, 5, 6, 7, 8, -2, 9]
    assert move([1, 2, -2, -3, 0, 3, 4], 2, -2) == [-2, 1, 2, -3, 0, 3, 4]


def solve(input: Input, second=False):
    xs = list(map(int, input.lines()))
    if second:
        xs = list(map(lambda n: n * 811589153, xs))
    ys = list(range(len(xs)))
    oys = ys[:]

    repeat = 1
    if second:
        repeat = 10
    for _ in range(repeat):
        for o in oys:
            i = ys.index(o)
            movev = xs[o]
            move(ys, i, movev)

    xs = [xs[i] for i in ys]
    s = 0
    i0 = xs.index(0)
    for i in range(1, 4):
        ni = (i0 + i * 1000) % len(ys)
        s += xs[ni]
    return s


def main():
    DAY_INPUT = "i20.txt"
    print("Example  1:", solve(Input(EXAMPLE)))
    print("Solution 1:", solve(Input(DAY_INPUT)))
    print("Example  2:", solve(Input(EXAMPLE), True))
    print("Solution 2:", solve(Input(DAY_INPUT), True))
    assert solve(Input(DAY_INPUT), True) == 6704537992933


if __name__ == "__main__":
    asserts()
    main()