Всегда появляется ошибка времени выполнения в пробке кода Google

Question

Я написал свой код в python -3 в google Code jam, к сожалению, я всегда получаю сообщение об ошибке выполнения, хотя код правильный и выполняется в блокноте jupyter.

I уже нашел один ответ на этот вопрос , но мне не ясно, как решить, как его реализовать, поэтому кто-то может дать мне решение этой проблемы.

Вот мой код -

import numpy as np
n = int(input())
a = []
b = []
t = []
for i in range(n):
    a.append(int(input()))
    for j in range(a[-1]):
        t.append(list(map(int, input().split())))

        b += t
        t = []
    x = a[0]
    k = 0


def cal(temp):
    row = 0
    k = 1
    for i in temp:
        dict = {}
        flag = 0
        for j in i:
            if j not in dict.keys():
                dict[j] = 1
                flag = 0

            else:
                flag = 1
                break

        if flag == 1:
            row += k
    return row


def func(temp):
    temp = np.array(temp)
    t = 0
    row = cal(temp)
    for i in range(len(temp)):
        t += temp[i][i]

    col = cal(temp.T)
    return t, row, col


for i in range(n):
    q = func(b[k:x])

    d = i + 1
    print("Case #{}: {} {} {}".format(d, q[0],q[1],q[2]))

    k = x
    x += a[i]

Answer 1

Вспомогательная функция для входов. Я не дам вам никакой логики c, но это поможет вам работать с системой Google Judge

def solve(matrix, size):
   return "{} {} {}".format(a, b, c)

for T in range(1, int(input()) + 1):
    size = int(input())
    matrix = [input().split(' ') for i in range(size)]
    print("Case #{}: {}".format(T, solve(matrix, size)))

Answer 2

numpy и другие внешние библиотеки недоступны. Кроме того, f strings не работают, как упомянуто.

Кроме того, во всем коде вы все равно не использовали никаких функций numpy, поэтому просто замените их на обычные python list, и они должны работать хорошо.

Удачи в GCJ ~