CSES - Datatähti 2023 alku - Results
Submission details
Task:Ruudukko
Sender:okkokko
Submission time:2022-11-04 22:59:34 +0200
Language:C++ (C++11)
Status:READY
Result:0
Feedback
groupverdictscore
#10
#20
#30
Test results
testverdicttimegroup
#10.00 s1, 2, 3details
#20.00 s1, 2, 3details
#30.00 s1, 2, 3details
#40.00 s2, 3details
#50.00 s2, 3details
#60.00 s2, 3details
#70.00 s3details
#80.00 s3details
#90.00 s3details

Code

#include <iostream>
#include <vector>
#include <unordered_set>
#include <set>
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;

int get_line_row(int y, int v)
{
    if (v == line_row_last[y])
    {
        return line_row[y];
    }
    line_row_last[y] = v;
    int s;
    s = (line_row[y] + line_row_added[y]) % DIVISOR;
    line_row_added[y] = 0;
    line_row[y] = s;

    return s;
}
int get_line_col(int x, int v)
{
    if (v == line_col_last[x])
    {
        return line_col[x];
    }
    line_col_last[x] = v;
    int s;
    s = (line_col[x] + line_col_added[x]) % DIVISOR;
    line_col_added[x] = 0;
    line_col[x] = s;

    return s;
}

int calculate_routes(int y, int x, int number)
{
    int value;
    if (number == 1)
    {
        value = 1;
    }
    else
    {
        value = (get_line_col(x, number) + get_line_row(y, number) + 1) % DIVISOR;
    }
    total += value;
    total %= DIVISOR;
    line_row_added[y] += value;
    line_row_added[y] %= DIVISOR;
    line_col_added[x] += value;
    line_col_added[x] %= DIVISOR;
    return value;
}

int main()
{
    int n;
    cin >> n;
    vector<vector<int>> grid;
    // grid.reserve(n * n);
    for (int i = 0; i < n; i++)
    {
        vector<int> b;
        grid.push_back(b);
        for (int j = 0; j < n; j++)
        {
            int a;
            cin >> a;
            grid[-1].push_back(a);
        }
    }

    vector<int> line_row(n, 0);
    vector<int> line_row_last(n, 1);
    vector<int> line_col(n, 0);
    vector<int> line_col_last(n, 1);
    vector<int> line_row_added(n, 0);
    vector<int> line_col_added(n, 0);

    vector<unordered_set<int>> exists_row;
    set<int> exists_all;
    for (int y = 0; y < n; y++)
    {
        unordered_set<int> s;
        for (int x = 0; x < n; x++)
        {
            s.insert(grid[y][x]);
            exists_all.insert(grid[y][x]);
        }
        exists_row.push_back(s);
    }

    for (auto &a : exists_all)
    {
        for (int y = 0; y < n; y++)
        {
            if (exists_row[y].find(a) != exists_row[y].end())
            {
                for (int x = 0; x < n; x++)
                {
                    if (grid[y][x] == a)
                    {
                        calculate_routes(y, x, a);
                    }
                }
            }
        }
    }
    cout << total << "\n";

    return 0;
}

Test details

Test 1

Group: 1, 2, 3

Verdict:

input
3
1 1 1
1 1 1
1 1 1

correct output
9

user output
(empty)

Test 2

Group: 1, 2, 3

Verdict:

input
3
1 2 3
6 5 4
7 8 9

correct output
135

user output
(empty)

Test 3

Group: 1, 2, 3

Verdict:

input
3
7 8 1
4 5 4
3 9 6

correct output
57

user output
(empty)

Test 4

Group: 2, 3

Verdict:

input
100
1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 ...

correct output
10000

user output
(empty)

Test 5

Group: 2, 3

Verdict:

input
100
1 2 3 4 5 6 7 8 9 10 11 12 13 ...

correct output
187458477

user output
(empty)

Test 6

Group: 2, 3

Verdict:

input
100
2995 8734 1018 2513 7971 5063 ...

correct output
964692694

user output
(empty)

Test 7

Group: 3

Verdict:

input
1000
1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 ...

correct output
1000000

user output
(empty)

Test 8

Group: 3

Verdict:

input
1000
1 2 3 4 5 6 7 8 9 10 11 12 13 ...

correct output
229147081

user output
(empty)

Test 9

Group: 3

Verdict:

input
1000
520283 805991 492643 75254 527...

correct output
951147313

user output
(empty)