Ruudukko

#include <bits/stdc++.h>
 
using namespace std;
 
const uint64_t AAAA = 1000000007;
 
uint64_t reitit[1000][1000];
uint64_t ruudukko[1000][1000];
pair<int, pair<int, int>> rows[1000][1000];
pair<int, pair<int, int>> cols[1000][1000];
uint64_t reititcols[1000][1000];
uint64_t reititrows[1000][1000];
int colsij[1000][1000];
int rowsij[1000][1000];
 
uint64_t maaralaske_reitit = 0;
uint64_t maaralaske_reitit_nopea = 0;
 
bool rowcolcmp(pair<int, pair<int, int>> i, pair<int, pair<int, int>> j)
{
    return i.first < j.first;
}
 
int main()
{
    int n;
    priority_queue<pair<int, pair<int, int>>> q;
    cin >> n;
    for (int i = 0; i < n; i++) {
        for (int j = 0; j < n; j++) {
            int arvo;
            //cin >> ruudukko[i][j];
            cin >> arvo;
            ruudukko[j][i] = arvo;
            q.push(make_pair(-arvo, make_pair(j, i)));
            rows[i][j] = make_pair(arvo, make_pair(j, i)); // :/
            cols[j][i] = make_pair(arvo, make_pair(j, i)); // ** :/ **
 
            reitit[i][j] = -1;
        }
    }
 
    for (int i = 0; i < n; i++) {
        sort(rows[i], rows[i]+n, rowcolcmp);
        sort(cols[i], cols[i]+n, rowcolcmp);
 
        for (int j = 0; j < n; j++) {
            pair<int, int> a = cols[i][j].second;
            pair<int, int> b = rows[i][j].second;
 
            colsij[a.first][a.second] = j;
            rowsij[b.first][b.second] = j;
        }
    }
 
    /*for (int j = 0; j < n; j++) {
        for (int i = 0; i < n; i++) {
            //cout << rows[i][j].first << endl;
            cout << i << ", " << j << ": " << colsij[i][j] << endl;
        }
    }*/
 
    //cout << "BEGIN --------------\n";
 
    uint64_t nreitit = 0;
 
    while (!q.empty()) {
        pair<int, pair<int, int>> a = q.top();
        int i = a.second.first;
        int j = a.second.second;
        int b = -a.first;
        q.pop();
        //cout << "------------- " << i << ", " << j << ":  " << b << endl;
 
        int acol = colsij[i][j] - 1;
        int edellinen = -1;
        uint64_t rr = 0;
        uint64_t rc = 0;
        for (int k = acol; k >= 0; k--) {
            pair<int, pair<int, int>> d = cols[i][k];
            int c = d.first;
 
            if (c == b) {
                continue;
            }
            
            if (edellinen != -1 && c < edellinen) {
                break;
            }
 
            edellinen = c;
            rr += reitit[d.second.first][d.second.second];
            rc += reititcols[d.second.first][d.second.second];
        }
 
        int arow = rowsij[i][j] - 1;
        edellinen = -1;
        uint64_t rr1 = 0;
        uint64_t rw = 0;
        //cout << "ROW " << arow << endl;
        for (int k = arow; k >= 0; k--) {
            pair<int, pair<int, int>> d = rows[j][k];
            int c = d.first;
 
            if (c == b) {
                continue;
            }
            
            if (edellinen != -1 && c < edellinen) {
                break;
            }
 
            edellinen = c;
            rr1 += reitit[d.second.first][d.second.second];
            rw += reititrows[d.second.second][d.second.first];
            //cout << "- " << d.second.first << ", " << d.second.second << endl;
            //cout << k << " " << rr1 << " " << rw << endl;
        }
 
        //cout << "rrrr: " << rr << " " << rc << " " << rr1 << " " << rw << endl;
 
        //uint64_t t = (rr + rr1 + rw + rc) % AAAA;
        uint64_t t = (rr + rr1 + 1) % AAAA;
        //cout << "--" << endl;
        //cout << t << endl;
        //reititrows[j][i] = t;
        //reititcols[i][j] = t;
        reititrows[j][i] = rr1;
        reititcols[i][j] = rr;
        //reititrows[j][i] = (rr1 + rw) % AAAA;
        //reititcols[i][j] = (rr + rc) % AAAA;
        reitit[i][j] = t + rw + rc;
        nreitit += t + rw + rc;
    }
 
    cout << nreitit % 1000000007 << endl;
    //cout << "Määrä laske_reitit: " << maaralaske_reitit << endl;
    //cout << "Määrä laske_reitit nopea: " << maaralaske_reitit_nopea << endl;
}

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: WRONG ANSWER

Test 6

Group: 2, 3

Verdict: WRONG ANSWER

Test 7

Group: 3

Verdict: TIME LIMIT EXCEEDED

Test 8

Group: 3

Verdict: WRONG ANSWER

Test 9

Group: 3

Verdict: TIME LIMIT EXCEEDED