Ruudukko

import random

class Array:
    def __init__(self, l):
        self.y_slices = l
        self.x_slices = [[sub[x] for sub in self.y_slices] for x in range(len(self.y_slices))]

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

def r(m, visited, x, y):
    current = m.y_slices[y][x]
    new = 0

    for axis_x, slice in [(False, m.x_slices[x]), (True, m.y_slices[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 = n**2
for y in range(n):
    for x in range(n):
        count += r(m, visited, x, y)
        
print(count % (10**9+7))

Test 1

Group: 1, 2, 3

Verdict: WRONG ANSWER

Test 2

Group: 1, 2, 3

Verdict: WRONG ANSWER

Test 3

Group: 1, 2, 3

Verdict: ACCEPTED

Test 4

Group: 2, 3

Verdict: WRONG ANSWER

Test 5

Group: 2, 3

Verdict: WRONG ANSWER

Test 6

Group: 2, 3

Verdict: WRONG ANSWER

Test 7

Group: 3

Verdict: TIME LIMIT EXCEEDED

Test 8

Group: 3

Verdict: TIME LIMIT EXCEEDED

Test 9

Group: 3

Verdict: TIME LIMIT EXCEEDED