Ruudukko

def mahdollisuudet(maarat, ruudukko, i, n):
    jakojaannos = i
    rivit = 0
    while jakojaannos >= n:
        jakojaannos -= n
        rivit += 1
    
    if rivit != n - 1:
        for a in range(1, n - rivit):
            if ruudukko[i + a*n] < ruudukko[i]:
                maarat[i + a*n] += 1
                lisattavat.append(i + a*n)
    if rivit != 0:
        for b in range(1, rivit + 1):
            if ruudukko[i - b*n] < ruudukko[i]:
                maarat[i - b*n] += 1
                lisattavat.append(i - b*n)
    if jakojaannos != 0:
        for c in range(1, jakojaannos + 1):
            if ruudukko[i - c] < ruudukko[i]:
                maarat[i - c] += 1
                lisattavat.append(i - c)
    if jakojaannos != n - 1:
        for d in range(1, n - jakojaannos):
            if ruudukko[i + d] < ruudukko[i]:
                maarat[i + d] += 1
                lisattavat.append(i + d)
        
    
 
if __name__ == "__main__":
    ruudukko = []
    maarat = []
    lisattavat = []
    n = int(input(""))
    summa = n*n
    
    #tehdään ruudukko
    for a in range(n):
        for b in input("").split(" "):
            b = int(b)
            ruudukko.append(b)
            maarat.append(0)
            
    
    for i in range(n*n):
        mahdollisuudet(maarat, ruudukko, i, n)
        
    while len(lisattavat) != 0:
        mahdollisuudet(maarat, ruudukko, lisattavat[0], n)
        lisattavat.pop(0)
        
    #for e in lisattavat:
    #    summa += maarat[e]
    print(sum(maarat) + summa)
        

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: 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