Ruudukko

#include <bits/stdc++.h>

using namespace std;

int main() {
    vector<pair<int,int>> v;
    map<int,pair<int,int>> m;
    int n, a, c, ans;
    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 j = 1; j <= n*n; j++) {
	    if (m.count(j) and v[i].first != m[j].second) {
		if (max(v[i].second,j)%n == min(v[i].second,j)%n) {
		    c = c+m[j].first;
		}
		else if (j > (v[i].second-(v[i].second%n)) and j <= (v[i].second-(v[i].second%n))+n and v[i].second%n != 0) {
		    c = c+m[j].first;
		}
		else if (v[i].second%n == 0 and j <= v[i].second and j > v[i].second - n) {
		    c = c+m[j].first;
		}
	    }
	}
	m[v[i].second] = {c,v[i].first};
	ans = ans + c;
    }
    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: TIME LIMIT EXCEEDED

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