Динамические массивы HackerRank

#python-3.x #dynamic-arrays

Вопрос:

Говоря об этом вопросе: https://www.hackerrank.com/challenges/three-month-preparation-kit-dynamic-array/problem?h_l=interviewamp;isFullScreen=falseamp;playlist_slugs[]=preparation-kitsamp;playlist_slugs[]=three-month-preparation-kitamp;playlist_slugs[]=three-month-week-five

Для тестового примера 0 все первые несколько запросов возвращают индекс, равный 1, но затем индекс изменяется на 0, что является пустым массивом, и я получаю ошибку времени выполнения по модулю 0.

Вероятно, та же проблема со всеми другими тестовыми примерами. Мой код прошел только тестовый пример 10. Я делаю что-то не так или это проблема с тестовыми примерами? Это гарантирует, что запрос 2 не будет пытаться запросить пустой массив или индекс.

Мой код:

     lastAnswer = 0
    arr = []
    [arr.append([]) for i in range(n)]
    answers = []
    
    for q in queries:
        if q[0] == 1:
            indx = (bool(q[1]) ^ bool(lastAnswer)) % n
            arr[indx].append(q[2])
            print('indx:', indx)
            print(arr)
        else:
            indx = (bool(q[1]) ^ bool(lastAnswer)) % n
            print('indx:', indx)
            lastAnswer = arr[indx][q[2] % len(arr[indx])]
            print('lastAsnwer: ', lastAnswer)
            print(arr)
            answers.append(lastAnswer)
    
    return answers