CSES - NOI 2024 - Results
Submission details
Task:Thin Ice
Sender:Victor Vatn
Submission time:2024-03-06 18:58:06 +0200
Language:C++17
Status:READY
Result:54
Feedback
groupverdictscore
#1ACCEPTED17
#2ACCEPTED12
#3ACCEPTED11
#40
#5ACCEPTED14
#60
Test results
testverdicttimegroup
#1ACCEPTED0.00 s1, 5, 6details
#2ACCEPTED0.00 s1, 5, 6details
#3ACCEPTED0.00 s1, 5, 6details
#4ACCEPTED0.00 s1, 5, 6details
#5ACCEPTED0.00 s1, 5, 6details
#6ACCEPTED0.00 s1, 5, 6details
#7ACCEPTED0.00 s1, 5, 6details
#8ACCEPTED0.00 s1, 5, 6details
#9ACCEPTED0.00 s1, 2, 5, 6details
#10ACCEPTED0.14 s2, 6details
#11ACCEPTED0.10 s2, 6details
#12ACCEPTED0.13 s2, 6details
#13ACCEPTED0.09 s2, 6details
#14ACCEPTED0.10 s2, 6details
#15ACCEPTED0.12 s2, 6details
#16ACCEPTED0.12 s2, 6details
#17ACCEPTED0.13 s2, 6details
#18ACCEPTED0.13 s2, 6details
#19ACCEPTED0.12 s2, 6details
#20ACCEPTED0.10 s2, 6details
#21ACCEPTED0.08 s2, 6details
#22ACCEPTED0.00 s3, 4, 5, 6details
#23ACCEPTED0.00 s3, 4, 5, 6details
#24ACCEPTED0.00 s3, 4, 5, 6details
#25ACCEPTED0.00 s3, 4, 5, 6details
#26ACCEPTED0.00 s3, 4, 5, 6details
#27ACCEPTED0.00 s3, 4, 5, 6details
#28--4, 6details
#29--4, 6details
#30--4, 6details
#31--4, 6details
#32--4, 6details
#33--4, 6details
#34--4, 6details
#35ACCEPTED0.01 s5, 6details
#36ACCEPTED0.01 s5, 6details
#37ACCEPTED0.01 s5, 6details
#38ACCEPTED0.01 s5, 6details
#39ACCEPTED0.01 s5, 6details
#40ACCEPTED0.01 s5, 6details
#41ACCEPTED0.01 s5, 6details
#420.82 s6details
#43--6details
#440.72 s6details
#45--6details
#46--6details
#470.99 s6details
#48--6details

Code

// #pragma GCC target ("avx,avx2,fma")
// #pragma GCC optimize ("Ofast,inline") // O1 - O2 - O3 - Os - Ofast
// #pragma GCC optimize ("unroll-loops")
#include <bits/stdc++.h>

using namespace std;

#define rep(i, a, b) for (ll i = (a); i < (b); ++i)
#define per(i, a, b) for (ll i = (b) - 1; i >= (a); --i)
#define trav(a, x) for (auto &a : x)

#define all(x) x.begin(), x.end()
#define sz(x) (ll)x.size()
#define pb push_back
#define debug(x) cout<<#x<<" = "<<x<<endl

#define umap unordered_map
#define uset unordered_set

typedef pair<int, int> ii;
typedef pair<int, ii> iii;
typedef vector<int> vi;
typedef vector<ii> vii;
typedef vector<vi> vvi;

typedef long long ll;
typedef pair<ll,ll> pll;
typedef pair<ll,pll> ppll;
typedef vector<ll> vll;
typedef vector<pll> vpll;
typedef vector<vll> vvll;

const ll INF = 1'000'000'007;

struct UnionFind {
    vll p,rank;
    ll n;
    UnionFind(ll N) : n(N) {
        p.resize(n);
        rep(i,0,n) p[i]=i;
        rank.assign(n,0);
    }

    ll find(ll i) { return p[i]==i?i:p[i]=find(p[i]); }
    bool unite(ll i,ll j) {
        i=find(i);
        j=find(j);
        if(i==j) return 0;

        if(rank[i]<rank[j]) swap(i,j);
        p[j]=i;
        if(rank[i]==rank[j]) ++rank[i];

        return 1;
    }
};

ll rows,cols,n;
vll vals;
vvll graph;

vll counts;

void dfs(ll u,ll p,ll mind,ll steps) {
    if(!steps) return;
    mind=min(mind,vals[u]);
    ++counts[mind];
    trav(v,graph[u]) if(v!=p) dfs(v,u,mind,steps-1);
}

int main() {
    cin.tie(0)->sync_with_stdio(0);
    cin.exceptions(cin.failbit);

    cin>>rows>>cols;
    n=rows*cols;
    graph.resize(n);

    ll maxd=0;
    rep(i,0,rows) rep(j,0,cols) {
        ll d;
        cin>>d;
        vals.pb(d);
        maxd=max(maxd,d);
    }

    vector<pair<ll,pll>> edges;
    rep(i,0,rows) rep(j,0,cols) {
        ll u=i*cols+j;
        if(i+1<rows) {
            ll v=(i+1)*cols+j;
            edges.pb({min(vals[u],vals[v]),{u,v}});
        }

        if(j+1<cols) {
            ll v=i*cols+j+1;
            edges.pb({min(vals[u],vals[v]),{u,v}});
        }
    }

    sort(all(edges),greater<>());
    UnionFind ufds(n);
    trav(edge,edges) {
        ll u,v;
        tie(u,v)=edge.second;
        if(ufds.unite(u,v)) {
            graph[u].pb(v);
            graph[v].pb(u);
            //cout<<vals[u]<<' '<<vals[v]<<endl;
        }
    }
    //cout<<endl;

    ll steps=n+1;
    if(n>1000) steps=5;
    ll ans=0;
    rep(row,0,rows) {
        rep(j,0,2) {
            counts.assign(maxd+1,0);
            ll col=0;
            if(j) col=cols-1;
            ll u=row*cols+col;
            dfs(u,u,INF,steps);
            ll curans=0;
            rep(i,0,maxd+1) curans+=min(i-curans,counts[i]);
            ans=max(ans,curans);
        }
    }

    rep(col,0,cols) {
        rep(j,0,2) {
            counts.assign(maxd+1,0);
            ll row=0;
            if(j) row=rows-1;
            ll u=row*cols+col;
            dfs(u,u,INF,steps);
            ll curans=0;
            rep(i,0,maxd+1) curans+=min(i-curans,counts[i]);
            ans=max(ans,curans);
        }
    }

    cout<<ans<<endl;
    return 0;
}

Test details

Test 1

Group: 1, 5, 6

Verdict: ACCEPTED

input
4 4
9 11 5 7
7 9 14 3
1 3 2 3
11 4 14 8

correct output
10

user output
10

Test 2

Group: 1, 5, 6

Verdict: ACCEPTED

input
5 3
10 7 11
8 5 5
12 9 10
3 13 9
...

correct output
10

user output
10

Test 3

Group: 1, 5, 6

Verdict: ACCEPTED

input
4 4
3 2 2 2
5 1 1 1
8 4 1 1
7 6 2 1

correct output
8

user output
8

Test 4

Group: 1, 5, 6

Verdict: ACCEPTED

input
3 5
1 11 1 3 5
4 12 7 8 7
13 14 14 9 4

correct output
14

user output
14

Test 5

Group: 1, 5, 6

Verdict: ACCEPTED

input
5 3
11 8 12
11 11 12
6 2 3
11 8 13
...

correct output
12

user output
12

Test 6

Group: 1, 5, 6

Verdict: ACCEPTED

input
3 5
14 14 14 14 14
14 14 14 14 14
14 14 14 14 14

correct output
14

user output
14

Test 7

Group: 1, 5, 6

Verdict: ACCEPTED

input
4 4
12 8 6 5
12 9 6 1
10 1 3 2
8 1 1 1

correct output
12

user output
12

Test 8

Group: 1, 5, 6

Verdict: ACCEPTED

input
4 4
8 3 15 14
10 12 2 4
5 16 9 6
7 13 1 11

correct output
13

user output
13

Test 9

Group: 1, 2, 5, 6

Verdict: ACCEPTED

input
4 4
4 1 1 2
3 5 4 2
2 2 1 2
1 4 3 4

correct output
4

user output
4

Test 10

Group: 2, 6

Verdict: ACCEPTED

input
10 20000
5 3 2 1 3 2 3 3 4 5 1 1 2 3 5 ...

correct output
5

user output
5

Test 11

Group: 2, 6

Verdict: ACCEPTED

input
10 20000
1 2 1 2 1 2 1 2 1 1 1 1 2 1 1 ...

correct output
3

user output
3

Test 12

Group: 2, 6

Verdict: ACCEPTED

input
10 20000
3 2 2 3 1 2 1 4 4 3 1 4 3 2 4 ...

correct output
5

user output
5

Test 13

Group: 2, 6

Verdict: ACCEPTED

input
20000 10
1 1 3 1 2 1 1 1 1 1
1 2 2 1 1 1 1 2 1 1
2 1 1 1 2 2 1 1 1 2
1 1 1 1 1 1 1 1 1 1
...

correct output
4

user output
4

Test 14

Group: 2, 6

Verdict: ACCEPTED

input
10 20000
1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 ...

correct output
3

user output
3

Test 15

Group: 2, 6

Verdict: ACCEPTED

input
10 20000
3 1 3 1 2 1 2 3 2 2 1 2 1 1 2 ...

correct output
3

user output
3

Test 16

Group: 2, 6

Verdict: ACCEPTED

input
10 20000
1 2 2 2 1 2 3 1 2 2 2 1 2 2 2 ...

correct output
4

user output
4

Test 17

Group: 2, 6

Verdict: ACCEPTED

input
10 20000
3 3 2 3 2 3 2 2 2 2 2 1 3 2 1 ...

correct output
4

user output
4

Test 18

Group: 2, 6

Verdict: ACCEPTED

input
10 20000
1 3 3 1 1 4 3 3 3 1 2 2 1 3 1 ...

correct output
5

user output
5

Test 19

Group: 2, 6

Verdict: ACCEPTED

input
7 28571
2 2 2 2 2 2 2 2 2 2 2 2 2 2 2 ...

correct output
3

user output
3

Test 20

Group: 2, 6

Verdict: ACCEPTED

input
28571 7
4 4 4 4 4 4 4
4 4 4 4 4 4 4
4 4 4 4 4 4 4
4 4 4 4 4 4 4
...

correct output
5

user output
5

Test 21

Group: 2, 6

Verdict: ACCEPTED

input
447 447
3 3 3 3 3 3 3 3 3 3 3 3 3 3 3 ...

correct output
3

user output
3

Test 22

Group: 3, 4, 5, 6

Verdict: ACCEPTED

input
1 100
46 23 23 55 37 17 30 32 25 71 ...

correct output
30

user output
30

Test 23

Group: 3, 4, 5, 6

Verdict: ACCEPTED

input
1 100
8 13 12 11 15 15 15 19 18 21 2...

correct output
76

user output
76

Test 24

Group: 3, 4, 5, 6

Verdict: ACCEPTED

input
1 100
94 94 94 95 95 91 97 100 92 98...

correct output
95

user output
95

Test 25

Group: 3, 4, 5, 6

Verdict: ACCEPTED

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

correct output
83

user output
83

Test 26

Group: 3, 4, 5, 6

Verdict: ACCEPTED

input
1 100
33 34 35 38 38 40 41 42 42 45 ...

correct output
95

user output
95

Test 27

Group: 3, 4, 5, 6

Verdict: ACCEPTED

input
1 100
57 1 80 39 18 63 30 86 85 55 8...

correct output
29

user output
29

Test 28

Group: 4, 6

Verdict:

input
1 200000
138736 14949 12344 104147 1333...

correct output
1806

user output
(empty)

Test 29

Group: 4, 6

Verdict:

input
1 200000
141245 141090 141163 141286 14...

correct output
155109

user output
(empty)

Test 30

Group: 4, 6

Verdict:

input
1 200000
102358 102469 102440 102402 10...

correct output
152388

user output
(empty)

Test 31

Group: 4, 6

Verdict:

input
1 200000
180410 160820 160820 180614 18...

correct output
160832

user output
(empty)

Test 32

Group: 4, 6

Verdict:

input
1 200000
198270 198270 198270 198270 19...

correct output
198270

user output
(empty)

Test 33

Group: 4, 6

Verdict:

input
1 200000
1 1 3 2 1 1 2 3 6 6 6 7 8 9 10...

correct output
199995

user output
(empty)

Test 34

Group: 4, 6

Verdict:

input
1 200000
14737 162555 44228 170991 1340...

correct output
1902

user output
(empty)

Test 35

Group: 5, 6

Verdict: ACCEPTED

input
31 32
669 792 226 189 860 737 291 83...

correct output
565

user output
565

Test 36

Group: 5, 6

Verdict: ACCEPTED

input
10 100
730 698 339 743 536 702 94 556...

correct output
529

user output
529

Test 37

Group: 5, 6

Verdict: ACCEPTED

input
32 31
633 613 618 605 635 638 668 67...

correct output
678

user output
678

Test 38

Group: 5, 6

Verdict: ACCEPTED

input
142 7
983 930 963 926 979 962 962
966 930 963 924 928 928 926
926 929 929 922 931 931 978
929 929 929 922 959 928 964
...

correct output
934

user output
934

Test 39

Group: 5, 6

Verdict: ACCEPTED

input
31 32
977 977 977 977 977 977 977 97...

correct output
977

user output
977

Test 40

Group: 5, 6

Verdict: ACCEPTED

input
50 20
1 27 14 23 38 48 56 3 12 9 6 2...

correct output
997

user output
997

Test 41

Group: 5, 6

Verdict: ACCEPTED

input
20 50
481 532 624 290 965 58 448 872...

correct output
504

user output
504

Test 42

Group: 6

Verdict:

input
447 447
6474 27185 108482 124481 16058...

correct output
88202

user output
24

Test 43

Group: 6

Verdict:

input
1000 200
27722 57131 197677 184858 9285...

correct output
89324

user output
(empty)

Test 44

Group: 6

Verdict:

input
447 447
70928 73154 72640 74764 75237 ...

correct output
181096

user output
23

Test 45

Group: 6

Verdict:

input
7 28571
193031 185883 171670 185794 17...

correct output
171680

user output
(empty)

Test 46

Group: 6

Verdict:

input
10 20000
191628 191628 191628 191628 19...

correct output
191628

user output
(empty)

Test 47

Group: 6

Verdict:

input
200 1000
3550 2640 3791 4248 4257 4504 ...

correct output
199997

user output
24

Test 48

Group: 6

Verdict:

input
1000 200
198379 62425 88013 50967 49098...

correct output
89131

user output
(empty)