Ruudukko

def mahdollisuudet(luku, sijainti, ruudukko, n):
    summa = 0
    jatkot = []
    
    
    if sijainti[1] != 0:
        for a in range(1, sijainti[1] + 1):
            if ruudukko[sijainti[0]][sijainti[1] - a] < luku:
                summa += 1
                jatkot.append([sijainti[0], sijainti[1] - a])
    
    elif sijainti[1] != n - 1:
        for d in range(1, n - sijainti[1]):
            if ruudukko[sijainti[0]][sijainti[1] + d] < luku:
                summa += 1
                jatkot.append([sijainti[0], sijainti[1] + d])
        
    
    if sijainti[0] != 0:
        for c in range(1, sijainti[0] + 1):
            if ruudukko[sijainti[0] - c][sijainti[1]] < luku:
                summa += 1
                jatkot.append([sijainti[0] - c, sijainti[1]])
    elif sijainti[0] != n - 1:
        for b in range(1, n - sijainti[0]):
            if ruudukko[sijainti[0] + b][sijainti[1]] < luku:
                summa += 1
                jatkot.append([sijainti[0] + b, sijainti[1]])
                
    if len(jatkot) != 0:
        return [summa, jatkot]
    else:return summa
    
    
    
 
if __name__ == "__main__":
    n = int(input(""))
    ruudukko = []
    summa = n**2
    tarkistettavat = []
    
    #tehdään ruudukko ja tarkistettavat
    for a in range(n):
        lista = input("").split(" ")
 
        ruudukko.append(lista)
        for d in range(n):
            tarkistettavat.append([a, d])
            
        
     
    while len(tarkistettavat) != 0:
        joku = mahdollisuudet(ruudukko[int(tarkistettavat[0][0])][int(tarkistettavat[0][1])], tarkistettavat[0], ruudukko, n)
        if type(joku) == list:
            summa += joku[0]
            for t in joku[1]:
                tarkistettavat.append(t)
        else:summa += joku
        tarkistettavat.pop(0)
        while summa >= 10**6 + 7:
            summa -= 10**6 + 7
        
    print(summa)
        
        

Test 1

Group: 1, 2, 3

Verdict: ACCEPTED

Test 2

Group: 1, 2, 3

Verdict: WRONG ANSWER

Test 3

Group: 1, 2, 3

Verdict: WRONG ANSWER

Test 4

Group: 2, 3

Verdict: ACCEPTED

Test 5

Group: 2, 3

Verdict: TIME LIMIT EXCEEDED

Test 6

Group: 2, 3

Verdict: TIME LIMIT EXCEEDED

Test 7

Group: 3

Verdict: TIME LIMIT EXCEEDED

Test 8

Group: 3

Verdict: TIME LIMIT EXCEEDED

Test 9

Group: 3

Verdict: TIME LIMIT EXCEEDED