CSES - Datatähti 2025 alku - Results
Submission details
Task:Niitty
Sender:Kemm1706
Submission time:2024-11-07 20:33:02 +0200
Language:C++ (C++11)
Status:READY
Result:58
Feedback
groupverdictscore
#1ACCEPTED4
#2ACCEPTED6
#3ACCEPTED10
#4ACCEPTED13
#5ACCEPTED25
#60
Test results
testverdicttimegroup
#1ACCEPTED0.00 s1, 2, 3, 4, 5, 6details
#2ACCEPTED0.01 s1, 2, 3, 4, 5, 6details
#3ACCEPTED0.00 s1, 2, 3, 4, 5, 6details
#4ACCEPTED0.00 s1, 2, 3, 4, 5, 6details
#5ACCEPTED0.00 s1, 2, 3, 4, 5, 6details
#6ACCEPTED0.00 s2, 3, 4, 5, 6details
#7ACCEPTED0.01 s2, 3, 4, 5, 6details
#8ACCEPTED0.00 s2, 3, 4, 5, 6details
#9ACCEPTED0.01 s2, 3, 4, 5, 6details
#10ACCEPTED0.01 s3, 4, 5, 6details
#11ACCEPTED0.01 s3, 4, 5, 6details
#12ACCEPTED0.00 s3, 4, 5, 6details
#13ACCEPTED0.01 s3, 4, 5, 6details
#14ACCEPTED0.01 s4, 5, 6details
#15ACCEPTED0.06 s4, 5, 6details
#16ACCEPTED0.01 s4, 5, 6details
#17ACCEPTED0.02 s4, 5, 6details
#18ACCEPTED0.03 s5, 6details
#19ACCEPTED0.25 s5, 6details
#20ACCEPTED0.01 s5, 6details
#21ACCEPTED0.04 s5, 6details
#22ACCEPTED0.14 s6details
#23--6details
#24ACCEPTED0.03 s6details
#25ACCEPTED0.45 s6details

Code

#include <bits/stdc++.h>
using namespace std;
typedef long long ll;
typedef vector <ll> vl;
typedef vector <vl> vvl;
typedef vector <vvl> table;



bool check(ll cal[][505][505], ll t, ll r, ll c, ll r2, ll c2)
{
    for(ll k = 0; k < t; k++)
    {
        if (cal[k][r2][c2] - cal[k][r - 1][c2] - cal[k][r2][c - 1] + cal[k][r - 1][c - 1] <= 0)
            return false;
    }
    return true;
}


bool findWindow(ll cal[][505][505], ll n, ll m, ll ns, ll i, ll j, ll &x, ll &y, ll rb)
{
    if (!check(cal, ns, i, j, n, m))
    {
        x = n + 1;
        y = m + 1;
        return false;
    }
    ll mid, lo = rb, step = 1, hi = rb - 1;
    while(hi + step <= n && !(check(cal, ns, i, j, hi + step, m)))
        step *= 2;
    hi += step;
    if (hi > n) hi = n;
    while(lo < hi)
    {
        mid = (lo + hi) / 2;
        if (!check(cal, ns, i, j, mid, m))
            lo = mid + 1;
        else
            hi = mid;
    }
    x = lo;
    lo = j;
    hi = m;
    while(lo < hi)
    {
        mid = (lo + hi) / 2;
        if (!check(cal, ns, i, j, x, mid))
            lo = mid + 1;
        else
            hi = mid;
    }
    y = lo;
    return true;
}


int main()
{
    ios_base::sync_with_stdio(false);
    cin.tie(0); cout.tie(0);

    ll n, i, j, x, y, area, t, k, m, ans = 0;
    string s;
    cin >> n;
    area = n * n;
    vl type(area);
    vvl field(n + 5, vl(n + 5)), rowb(n + 5, vl(n + 5));

    for(i = 1; i <= n; i++)
    {
        cin >> s;
        for(j = 1; j <= n; j++)
        {
            field[i][j] = s[j - 1] - 'A';
            type[i * n - n + j - 1] = field[i][j];
        }
    }
    sort(type.begin(), type.end());
    t = unique(type.begin(), type.end()) - type.begin();
    map <ll, ll> mp;
    for(k = 0; k < t; k++)
        mp[type[k]] = k;

    /*for(i = 1; i <= n; i++)
    {
        for(j = 1; j <= n; j++)
            cerr << mp[field[i][j]] << " ";
        cerr << "\n";
    }*/

    //cerr << "!!!";

    // table cal(t, vvl(n + 5, vl(n + 5, 0)));
    static ll cal[26][505][505];
    for(i = 1; i <= n; i++)
        for(j = 1; j <= n; j++)
        {
            cal[mp[field[i][j]]][i][j] = 1;
            for(k = 0; k < t; k++)
                cal[k][i][j] += cal[k][i - 1][j] + cal[k][i][j - 1] - cal[k][i - 1][j - 1];
        }


//    for(i = 1; i <= n; i++)
//        for(j = 1; j <= n; j++)
//        {
//            m = n;
//            while(findWindow(cal, n, m, t, i, j, x, y))
//            {
//                ans += (m - y + 1) * (n - x + 1);
//                m = y - 1;
//                //cerr << i << " " << j << " " << x << " " << y << "\n";
//            }
//        }


    for(i = 1; i <= n; i++)
        fill(rowb[i].begin() + 1, rowb[i].end(), i);
    for(i = 1; i <= n; i++)
    {
        for(j = 1; j <= n; j++)
        {
            m = n;
            if(rowb[i][j] > n || !findWindow(cal, n, m, t, i, j, x, y, rowb[i][j]))
            {
                if (j < n) rowb[i][j + 1] = n + 1;
                if (i < n) rowb[i + 1][j] = n + 1;
                j = n;
                continue;
            }
            if (i < n)
                rowb[i + 1][j] = max(x, i + 1);
            ans += (m-y+1)*(n-x+1);
            m = y-1;
            while(findWindow(cal, n, m, t, i, j, x, y, rowb[i][j]))
            {
                ans += (m - y + 1) * (n - x + 1);
                m = y - 1;
            }
        }
    }


    /*for(k = 0; k < t; k++)
    {
        cerr << k << ":\n";
        for(i = 0; i <= n; i++)
        {
            for(j = 0; j <= n; j++)
                cerr << cal[k][i][j] << " ";
            cerr << "\n";
        }
    }*/

    cout << ans;

    return 0;
}

Test details

Test 1

Group: 1, 2, 3, 4, 5, 6

Verdict: ACCEPTED

input
10
TNCTNPNTPC
NPPNTNTPTP
NTNTTCNTCT
NPCPNPPNTT
...

correct output
2035

user output
2035

Test 2

Group: 1, 2, 3, 4, 5, 6

Verdict: ACCEPTED

input
10
NFWQLWNWYS
DZOQJVXFPJ
CNHXPXMCQD
QRTBVNLTQC
...

correct output
9

user output
9

Test 3

Group: 1, 2, 3, 4, 5, 6

Verdict: ACCEPTED

input
10
XXXXXXXXXX
XXXXXXXXXX
XXXXXXXXXX
XXXXXXXXXX
...

correct output
3025

user output
3025

Test 4

Group: 1, 2, 3, 4, 5, 6

Verdict: ACCEPTED

input
10
FFFFFFFFFF
FFFFFCFFFF
FFFFFFJFFF
FFFFFFFFFF
...

correct output
12

user output
12

Test 5

Group: 1, 2, 3, 4, 5, 6

Verdict: ACCEPTED

input
1
X

correct output
1

user output
1

Test 6

Group: 2, 3, 4, 5, 6

Verdict: ACCEPTED

input
20
BBCBUBOUOBBCUUBBCOUO
BOUCOOCUBCOOOCOBOCUO
UCCUUUOBCOCBCBUBUCOO
BUOBUCUCUOOBCOOUBUOO
...

correct output
38724

user output
38724

Test 7

Group: 2, 3, 4, 5, 6

Verdict: ACCEPTED

input
20
CBGLSHGZHYZDWBNDBJUG
SMUXOJQYPXZDTMJUIWOJ
XIDSTNBGHKRKOVUVMINB
MTQGCFRUHQKALXRNCQGS
...

correct output
8334

user output
8334

Test 8

Group: 2, 3, 4, 5, 6

Verdict: ACCEPTED

input
20
KKKKKKKKKKKKKKKKKKKK
KKKKKKKKKKKKKKKKKKKK
KKKKKKKKKKKKKKKKKKKK
KKKKKKKKKKKKKKKKKKKK
...

correct output
44100

user output
44100

Test 9

Group: 2, 3, 4, 5, 6

Verdict: ACCEPTED

input
20
AAAAAAAAXAAAAAAAAAAA
AAAWAAAAAAAAAAAAAOAA
AAAAAAAAAAAAAAAAAPAA
AAAAAAAAKAAAAAAAAAAZ
...

correct output
18

user output
18

Test 10

Group: 3, 4, 5, 6

Verdict: ACCEPTED

input
50
GRGREEEGREGXRXXEGXXREXGRRRGRRR...

correct output
1584665

user output
1584665

Test 11

Group: 3, 4, 5, 6

Verdict: ACCEPTED

input
50
AITIISJUHCCRZNKSDCNQKYSQRINFWJ...

correct output
1077746

user output
1077746

Test 12

Group: 3, 4, 5, 6

Verdict: ACCEPTED

input
50
OOOOOOOOOOOOOOOOOOOOOOOOOOOOOO...

correct output
1625625

user output
1625625

Test 13

Group: 3, 4, 5, 6

Verdict: ACCEPTED

input
50
FFFFFFFFFFFFFFFFFFFFFFFFFFFFFF...

correct output
1680

user output
1680

Test 14

Group: 4, 5, 6

Verdict: ACCEPTED

input
100
NNCMDCDDCCNNNDNCMMNCDCDCCDCDNM...

correct output
25325366

user output
25325366

Test 15

Group: 4, 5, 6

Verdict: ACCEPTED

input
100
LIMQQIHASECROEVILNVULGWZJPPKOG...

correct output
22342463

user output
22342463

Test 16

Group: 4, 5, 6

Verdict: ACCEPTED

input
100
TTTTTTTTTTTTTTTTTTTTTTTTTTTTTT...

correct output
25502500

user output
25502500

Test 17

Group: 4, 5, 6

Verdict: ACCEPTED

input
100
QXQQQQQQQQQQQQQQQQQQQQQQQQQQQQ...

correct output
25650

user output
25650

Test 18

Group: 5, 6

Verdict: ACCEPTED

input
200
NAANANMMKNKKAKMKMAKNKMNKMMNNAA...

correct output
403292767

user output
403292767

Test 19

Group: 5, 6

Verdict: ACCEPTED

input
200
OMYWATTLURKQPTKEFMGGYAOONXWVSC...

correct output
388111321

user output
388111321

Test 20

Group: 5, 6

Verdict: ACCEPTED

input
200
CCCCCCCCCCCCCCCCCCCCCCCCCCCCCC...

correct output
404010000

user output
404010000

Test 21

Group: 5, 6

Verdict: ACCEPTED

input
200
LLLLLLLLLLLLLLLLLHLLLLLLLLLLLL...

correct output
14159445

user output
14159445

Test 22

Group: 6

Verdict: ACCEPTED

input
500
VVHWVUHVHUWWWVUUUWVUUHUUWHWUVW...

correct output
15683003812

user output
15683003812

Test 23

Group: 6

Verdict:

input
500
OIMZGEQSBMBDSDXSWRFNKSGFEBBTJE...

correct output
15575906951

user output
(empty)

Test 24

Group: 6

Verdict: ACCEPTED

input
500
IIIIIIIIIIIIIIIIIIIIIIIIIIIIII...

correct output
15687562500

user output
15687562500

Test 25

Group: 6

Verdict: ACCEPTED

input
500
WWWWWWWWWWWWWWWWWWWWWWWWWWWWWW...

correct output
3058970930

user output
3058970930