Ruudukko

#include <iostream>
#include <vector>
 
 
int main() {
 
    unsigned int n;
    unsigned int x;
    unsigned int routes_in_mod = 0;     // MODULO: x % 1000000007
    std::cin >> n;
 
    if (n < 2) { std::cout << 1 << '\n'; return 0; }
    
    unsigned int ROWS[1000]    = { };
    unsigned int COLUNMS[1000] = { };
 
    unsigned int TEMP_ROWS[1000]    = { };
    unsigned int TEMP_COLUNMS[1000] = { };
    
    // POINTER LIST INITIALIZATION
    std::vector<unsigned int>* POINTER_LIST = new std::vector<unsigned int>[n*n]();
 
    // GET DATA LOCATION INTO POINTER LIST
    for (unsigned int i = 0; i < n*n; i++)
    {
        std::cin >> x;
        POINTER_LIST[x-1].push_back(i);
    }
 
    // DO the magic
    for (unsigned int i = 0; i < POINTER_LIST[0].size(); i++)
    {
        ROWS[POINTER_LIST[0][i] % n] += 1;
        COLUNMS[(unsigned int)(POINTER_LIST[0][i] / n)] += 1;
 
        TEMP_ROWS[POINTER_LIST[0][i] % n] += 1;
        TEMP_COLUNMS[(unsigned int)(POINTER_LIST[0][i] / n)] += 1;
    }
 
    unsigned int c = 0;
    for (unsigned int i = 1; i < n*n; i++)
    {
        for (unsigned int j = 0; j < POINTER_LIST[i].size(); j++)
        {
            c = (ROWS[POINTER_LIST[i][j] % n] + COLUNMS[(unsigned int)(POINTER_LIST[i][j] / n)]) % 1000000007;
            routes_in_mod = routes_in_mod + c;
            TEMP_ROWS[POINTER_LIST[i][j] % n] += c + 1;
            TEMP_COLUNMS[(unsigned int)(POINTER_LIST[i][j] / n)] += c + 1;
        }
 
        for (unsigned int j = 0; j < 1000; j++)
        {
            ROWS[j] = TEMP_ROWS[j];
            COLUNMS[j] = TEMP_COLUNMS[j];
        }
    }
    routes_in_mod = ( routes_in_mod + n*n) % 1000000007;
    
    // Deletion
    delete[] POINTER_LIST;
 
    // Answer
    std::cout << routes_in_mod << "\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: WRONG ANSWER

Test 6

Group: 2, 3

Verdict: WRONG ANSWER

Test 7

Group: 3

Verdict: ACCEPTED

Test 8

Group: 3

Verdict: WRONG ANSWER

Test 9

Group: 3

Verdict: WRONG ANSWER