Ruudukko

from functools import cache

# clean version
def Ruudukko():
    global R1, R2
    # käsittelee syötteen
    koko = int(input())
    kokorange = range(koko)
    ruudukko1 = [[int(s) for s in input().split(" ")] for _ in kokorange]
    ruudukko2 = [[rivi[n] for rivi in ruudukko1] for n in kokorange]
    R1 = ruudukko1
    R2 = ruudukko2
    # käy läpi kaikki vaihtoehdot rekursiivisesti
    polut = 0
    for c1 in kokorange:
        for c2 in kokorange:
            polut += rekursiivinen_reitti(c1, c2)
    # yhden pituiset reitit
    polut += koko * koko
    # palauttaa syötteen
    return polut % (10**9 + 7)

@cache
def rekursiivinen_reitti(c1: int, c2: int) -> int:
    polut = 0
    num = R1[c1][c2]
    # kaikki vaihtoehdot vaakarivillä
    i = 0
    for n in R1[c1]:
        # löytyi reitti riviltä
        if n < num:
            polut += rekursiivinen_reitti(c1, i) + 1
        i += 1
    # sama pystyrivillä
    i = 0
    for n in R2[c2]:
        if n < num:
            polut += rekursiivinen_reitti(i, c2) + 1
        i += 1
    # palauttaa polut
    return polut

print(Ruudukko())

Test 1

Group: 1, 2, 3

Verdict: ACCEPTED

Test 2

Group: 1, 2, 3

Verdict: ACCEPTED

Test 3

Group: 1, 2, 3

Verdict: ACCEPTED

Test 4

Group: 2, 3

Verdict: ACCEPTED

Test 5

Group: 2, 3

Verdict: ACCEPTED

Test 6

Group: 2, 3

Verdict: ACCEPTED

Test 7

Group: 3

Verdict: TIME LIMIT EXCEEDED

Test 8

Group: 3

Verdict: RUNTIME ERROR

Traceback (most recent call last):
  File "/box/input/code.py", line 43, in <module>
    p...

Test 9

Group: 3

Verdict: TIME LIMIT EXCEEDED