Ruudukko

import random

class Array:
    def __init__(self, l):
        self.l = l

    def index(self, x=None, y=None):
        if x == None and y == None:
            return None
        elif x != None and y != None:
            return self.l[y][x]
        elif x == None:
            return self.l[y]
        elif y == None:
            return [sub[x] for sub in self.l]
    
    def __str__(self):
        s = ""

        for sub in self.l:
            s += " ".join(map(str, sub)) + "\n"

        return s
    

        

n = int(input())
m = Array([list(map(int, input().split())) for _ in range(n)])
#m = Array([[random.randint(1, n**2) for __ in range(n)] for _ in range(n)])
visited = {}
#print(m)

def r(m, visited, x, y):
    current = m.index(x, y)
    new = 0

    for axis_x, slice in [(False, m.index(x=x)), (True, m.index(y=y))]:
        for i in range(len(slice)):
            if slice[i] < current:
                new += 1
                if axis_x:
                    if (i, y) in visited:
                        new += visited[(i, y)]
                    else:
                        new += r(m, visited, i, y)
                else:
                    if (x, i) in visited:
                        new += visited[(x, i)]
                    else:
                        new += r(m, visited, x, i)
    
    visited[(x, y)] = new
    return new


count = 0
for y in range(n):
    for x in range(n):
        count += 1
        count += r(m, visited, x, y)
        
print(count % (10**9+7))

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

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

Test 9

Group: 3

Verdict: TIME LIMIT EXCEEDED