Ruudukko

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

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