Ruudukko

#include <iostream>
#include <vector>
#include <unordered_set>
#include <set>
#include <bits/stdc++.h>
using namespace std;
 
const int DIVISOR = 1000000007;
vector<int> line_row;
vector<int> line_row_last;
vector<int> line_col;
vector<int> line_col_last;
vector<int> line_row_added;
vector<int> line_col_added;
int total = 0;
struct hashFunction
{
    size_t operator()(const tuple<int, int, int> &x) const
    {
        return get<0>(x) ^ get<1>(x) ^ get<2>(x);
    }
};
 
void calculate_routes(int number, int x, int y)
{
    if (number != line_col_last[x])
    {
        line_col_last[x] = number;
        line_col[x] = (line_col[x] + line_col_added[x]) % DIVISOR;
        line_col_added[x] = 0;
    }
    if (number != line_row_last[y])
    {
        line_row_last[y] = number;
        line_row[y] = (line_row[y] + line_row_added[y]) % DIVISOR;
        line_row_added[y] = 0;
    }
 
    int value = (line_col[x] + line_row[y] + 1) % DIVISOR;
    total = (total + value) % DIVISOR;
    line_row_added[y] = (line_row_added[y] + value) % DIVISOR;
    line_col_added[x] = (line_col_added[x] + value) % DIVISOR;
}
bool cmp(tuple<int, int, int> x, tuple<int, int, int> y)
{
    return get<0>(x) > get<0>(y);
}
 
int main()
{
    int n;
    cin >> n;
    unordered_set<tuple<int, int, int>, hashFunction> values_coords;
    values_coords.reserve(n * n);
    for (int i = 0; i < n; i++)
    {
        for (int j = 0; j < n; j++)
        {
            int a;
            cin >> a;
            values_coords.insert(make_tuple(a, j, i));
        }
    }
 
    line_row = vector<int>(n, 0);
    line_row_last = vector<int>(n, 1);
    line_col = vector<int>(n, 0);
    line_col_last = vector<int>(n, 1);
    line_row_added = vector<int>(n, 0);
    line_col_added = vector<int>(n, 0);
 
    vector<tuple<int, int, int>> v(values_coords.begin(), values_coords.end());
    sort(v.begin(), v.end(), cmp);
 
    vector<tuple<int, int, int>>::iterator itr;
    for (itr = v.begin(); itr != v.end(); itr++)
    {
        calculate_routes(get<0>(*itr), get<1>(*itr), get<2>(*itr));
    }
    cout << total << "\n";
 
    return 0;
}

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: ACCEPTED

Test 9

Group: 3

Verdict: ACCEPTED