DyanamicArray.py

#!/bin/python3

import math
import os
import random
import re
import sys

#
# Complete the 'dynamicArray' function below.
#
# The function is expected to return an INTEGER_ARRAY.
# The function accepts following parameters:
#  1. INTEGER n
#  2. 2D_INTEGER_ARRAY queries
#

def dynamicArray(n, queries):
    # Write your code here
    lastAnswer = 0
    result = []
    seqList = [[] for i in range(n)] 
    for i in range(len(queries)):
        if queries[i][0] == 1:
            seqList[(queries[i][1] ^ lastAnswer) % n].append(queries[i][2])
        elif queries[i][0] == 2:
            lastAnswer = seqList[(queries[i][1] ^ lastAnswer) % n][queries[i][2] % len(seqList[(queries[i][1] ^ lastAnswer) % n])]
            result.append(lastAnswer)
    return result




if __name__ == '__main__':
    fptr = open(os.environ['OUTPUT_PATH'], 'w')

    first_multiple_input = input().rstrip().split()

    n = int(first_multiple_input[0])

    q = int(first_multiple_input[1])

    queries = []

    for _ in range(q):
        queries.append(list(map(int, input().rstrip().split())))

    result = dynamicArray(n, queries)

    fptr.write('\n'.join(map(str, result)))
    fptr.write('\n')

    fptr.close()