CSES - Datatähti 2023 alku - Results
Submission details
Task:Ruudukko
Sender:stpn129
Submission time:2022-11-13 18:34:28 +0200
Language:C++11
Status:READY
Result:100
Feedback
groupverdictscore
#1ACCEPTED28
#2ACCEPTED33
#3ACCEPTED39
Test results
testverdicttimegroup
#1ACCEPTED0.00 s1, 2, 3details
#2ACCEPTED0.00 s1, 2, 3details
#3ACCEPTED0.00 s1, 2, 3details
#4ACCEPTED0.01 s2, 3details
#5ACCEPTED0.01 s2, 3details
#6ACCEPTED0.01 s2, 3details
#7ACCEPTED0.17 s3details
#8ACCEPTED0.21 s3details
#9ACCEPTED0.28 s3details

Code

#ifdef LOCAL
#include <iostream>
#include <algorithm>
#include <vector>
#include <set>
#include <unordered_set>
#include <unordered_map>
#include <numeric>
#include <cmath>
#include <deque>
#include <random>
#include <bitset>
#include <map>
#include <cassert>
#include <complex>
#include <cmath>
#include <time.h>
#else
#include "bits/stdc++.h"
#endif

using namespace std;
using cd = complex<double>;

#pragma GCC optimize("Ofast")
#pragma GCC optimize("unroll-loops")

#define ll int64_t
#define ld long double
#define all(x) (x).begin(), (x).end()
#define rall(x) (x).rbegin(), (x).rend()
#define x first
#define y second

#define int ll
#define double ld
#define endl '\n'

const int inf = 1e9;
const ll INF = 1e18;

const int mod = 1e9 + 7;

void solve() {
    int n;
    cin >> n;
    vector<pair<int, pair<int, int>>> arr;
    for (int i = 1; i <= n; ++i) {
        for (int j = 1; j <= n; ++j) {
            int x;
            cin >> x;
            arr.push_back({x, {i, j}});
        }
    }

    sort(all(arr));

    vector<int> a(n + 1), b(n + 1);

    vector<int> kek(n * n);

    for (int i = 0; i < n * n;) {
        int j = i;
        while (j < n * n && arr[j].x == arr[i].x) {
            ++j;
        }

        for (int x = i; x < j; ++x) {
            kek[x] = a[arr[x].y.x] + b[arr[x].y.y] + 1;
        }

        for (int x = i; x < j; ++x) {
            a[arr[x].y.x] = (a[arr[x].y.x] + kek[x]) % mod;
            b[arr[x].y.y] = (b[arr[x].y.y] + kek[x]) % mod;
        }

        i = j;
    }

    int ans = 0;
    for (int i = 0; i < n * n; ++i)
        ans = (ans + kek[i]) % mod;

    cout << ans << endl;
}

signed main() {
#ifndef LOCAL
    cin.tie(NULL);
    cout.tie(NULL);
    ios_base::sync_with_stdio(false);
#endif


    cout.precision(16);
    cout << fixed;

    int t = 1;
    // cin >> t;
    while (t--) {
        solve();
    }
}

Test details

Test 1

Group: 1, 2, 3

Verdict: ACCEPTED

input
3
1 1 1
1 1 1
1 1 1

correct output
9

user output
9

Test 2

Group: 1, 2, 3

Verdict: ACCEPTED

input
3
1 2 3
6 5 4
7 8 9

correct output
135

user output
135

Test 3

Group: 1, 2, 3

Verdict: ACCEPTED

input
3
7 8 1
4 5 4
3 9 6

correct output
57

user output
57

Test 4

Group: 2, 3

Verdict: ACCEPTED

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

correct output
10000

user output
10000

Test 5

Group: 2, 3

Verdict: ACCEPTED

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

correct output
187458477

user output
187458477

Test 6

Group: 2, 3

Verdict: ACCEPTED

input
100
2995 8734 1018 2513 7971 5063 ...

correct output
964692694

user output
964692694

Test 7

Group: 3

Verdict: ACCEPTED

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

correct output
1000000

user output
1000000

Test 8

Group: 3

Verdict: ACCEPTED

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

correct output
229147081

user output
229147081

Test 9

Group: 3

Verdict: ACCEPTED

input
1000
520283 805991 492643 75254 527...

correct output
951147313

user output
951147313