Ruudukko

#include <bits/stdc++.h>
 
using namespace std;
 
int main() {
    vector<pair<long long,long long>> v;
    map<long long,pair<long long,long long>> m;
    long long n, a, c, j, ans;
    j = 0;
    ans = 0;
    c = 0;
    cin >> n;
    for (int i = 1; i <= n*n; i++) {
	cin >> a;
	v.push_back({a,i});
    }
    sort(v.begin(), v.end());
    for (int i = 0; i < n*n; i++) {
	c = 1;
	for (int x = 1; x <= n; x++) {
	    j = (x-1)*n+(v[i].second%n);
	    if (m.count(j) and v[i].first != m[j].second) {
		c = c+m[j].first;
	    }
	}
	for (int x = 1; x <= n; x++) {
	    if (v[i].second%n != 0) {
		j = (v[i].second-v[i].second%n)+x;
	    }
	    else {
		j = (v[i].second-n)+x;
	    }
	    if (m.count(j) and v[i].first != m[j].second) {
		c = c+m[j].first;
	    }
	}
	if (c < 0) {
	    while (1 == 1) {

	    }
	}
	m[v[i].second] = {c,v[i].first};
	ans = ans + c;
	ans = ans%1000000007;
    }
    cout << ans;
}

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