Ruudukko

total = 0
 
by_number = {}
 
by_row = {}
by_column = {}
 
size = int(input())
 
for y in range(size):
	# it's a square, we can do this
	by_row[y] = 0
	by_column[y] = 0
 
	data = input().split(" ")
	for x in range(size):
		num = data[x]
 
		if num not in by_number:
			by_number[num] = set()
 
		by_number[num].add((x, y))
 
for num in range(1, size**2 + 1):
	if str(num) not in by_number:
		continue
 
	found_for_rows = {}
	found_for_columns = {}
 
	for i in range(size):
		found_for_rows[i] = 0
		found_for_columns[i] = 0
 
	for x, y in by_number[str(num)]:
		for_current = 1 + by_row[y] + by_column[x]
 
		found_for_rows[y] += for_current
		found_for_columns[x] += for_current
 
		total += for_current
 
		#print(num, x, y, for_current)
	
	for key, value in found_for_rows.items():
		by_row[key] += value
	for key, value in found_for_columns.items():
		by_column[key] += value
 
#print(found)
 
print(total % (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: ACCEPTED

Test 6

Group: 2, 3

Verdict: ACCEPTED

Test 7

Group: 3

Verdict: TIME LIMIT EXCEEDED

Test 8

Group: 3

Verdict: TIME LIMIT EXCEEDED

Test 9

Group: 3

Verdict: TIME LIMIT EXCEEDED