oj.uz

Submission #291212

# Submission time Handle Problem Language Result Execution time Memory
291212 2020-09-04T21:48:17 Z thebes Rectangles (IOI19_rect) C++14
100 / 100
 2454 ms 863948 KB 
#include "rect.h"
#include <bits/stdc++.h>
using namespace std;

typedef long long ll;
typedef vector<int> vi;
typedef pair<int,int> pii;
#define F first
#define S second

const int MN = 2505, MM = 15e6;
int N, M, C, i, j, x, y, h[MM], hp[MM], cnt, pre[MN], ds[MN], nxt[MN]; ll ans;
pii bnd[MM]; vi vec[MN], adj[MM], hid[MN];
vector<pii> evt[MN];
pii hist[MN][MN];
stack<pii> stk;
vector<pii> st[MN];

inline void process(vi vec,int x,bool col){
    C = (int)vec.size();
    for(int i=0;i<=C;i++) st[i].clear();
    while(stk.size()) stk.pop();
    int flg;
    for(int i=0;i<C;i++){
        flg = 0;
        while(stk.size()){
            if(stk.top().S+1<i&&!flg){
                if(col) evt[i-1].push_back({x,stk.top().S+1});
                else{
                    bnd[++cnt] = {stk.top().S+1,i-1};
                    h[cnt] = x;
                    st[stk.top().S+1].push_back({i-1,cnt});
                }
            }
            if(stk.top().F==vec[i]) flg=1;
            if(stk.top().F<=vec[i]) stk.pop();
            else break;
        }
        stk.push({vec[i],i});
    }
    while(stk.size()) stk.pop();
    for(int i=0;i<=C;i++){
        reverse(st[i].begin(),st[i].end());
        for(auto v : st[i]){
            while(stk.size()){
                if(stk.top().F<v.F) stk.pop();
                else break;
            }
            if(stk.size()){
                adj[stk.top().S].push_back(v.S);
                hp[v.S] = 1;
            }
            stk.push(v);
        }
    }
}

vi res;

inline void mrg(int x,int y){
    if(x^y){
        res.clear();
        if(vec[x].empty()||vec[y].empty()){
            vec[x].clear(); vec[y].clear();
            return;
        }
        if(vec[x].size()>vec[y].size()) swap(x,y);
        for(int i=0,j=0;i<(int)vec[x].size();i++){
            while(j+1<(int)vec[y].size()&&vec[y][j]<vec[x][i]) j++;
            if(j<vec[y].size()&&vec[x][i]==vec[y][j]) res.push_back(vec[x][i]);
        }
        vec[min(x,y)].clear();
        vec[max(x,y)]=res;
    }
}

int fnd(int x){return ds[x]=ds[x]==x?x:fnd(ds[x]);}

void dfs(int n){
    for(auto v : adj[n]) dfs(v);
    for(auto v : adj[n]) mrg(bnd[v].S, bnd[n].S);
    int it = fnd(bnd[n].F);
    while(it<M&&it<=bnd[n].S){
        mrg(it,bnd[n].S);
        ds[fnd(it)]=fnd(it+1);
        it = fnd(it);
    }
    int p = upper_bound(vec[bnd[n].S].begin(),vec[bnd[n].S].end(),h[n])-vec[bnd[n].S].begin();
    ans += vec[bnd[n].S].size()-p;
}

long long count_rectangles(vector<vector<int>> a){
    vi col;
	N = a.size(), M = a[0].size();
	memset(hist,-1,sizeof(hist));
	for(i=0;i<M;i++){
        col.clear();
        for(j=0;j<N;j++) col.push_back(a[j][i]);
        process(col,i,1);
	}
	memset(hist,-1,sizeof(hist));
	for(i=0;i<N;i++)
        process(a[i],i,0);
	for(i=1;i<=cnt;i++){
        hid[h[i]].push_back(i);
	}
	memset(hist,-1,sizeof(hist));
	for(i=0;i<N;i++){
        for(j=0;j<M;j++) vec[j].clear(), ds[j]=j;
        for(auto v : evt[i]) vec[v.F].push_back(v.S);
        for(j=0;j<M;j++) sort(vec[j].begin(),vec[j].end());
        for(auto v : hid[i]){
            if(hist[bnd[v].F][bnd[v].S].F==i) h[v] = hist[bnd[v].F][bnd[v].S].S+1;
            else h[v] = 1;
            hist[bnd[v].F][bnd[v].S]={i+1,h[v]};
            h[v] = i-h[v];
        }
        for(auto v : hid[i]){
            if(hp[v]) continue;
            dfs(v);
        }
	}
	return ans;
}

Compilation message

rect.cpp: In function 'void mrg(int, int)':
rect.cpp:70:17: warning: comparison of integer expressions of different signedness: 'int' and 'std::vector<int>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
   70 |             if(j<vec[y].size()&&vec[x][i]==vec[y][j]) res.push_back(vec[x][i]);
      |                ~^~~~~~~~~~~~~~

Subtask #1
8.0 / 8.0

# Verdict Execution time Memory Grader output
1 Correct 256 ms 401912 KB Output is correct
2 Correct 260 ms 402040 KB Output is correct
3 Correct 260 ms 402048 KB Output is correct
4 Correct 261 ms 402040 KB Output is correct
5 Correct 259 ms 401912 KB Output is correct
6 Correct 258 ms 402040 KB Output is correct
7 Correct 259 ms 401912 KB Output is correct
8 Correct 259 ms 401912 KB Output is correct
9 Correct 262 ms 401912 KB Output is correct
10 Correct 262 ms 401912 KB Output is correct
11 Correct 259 ms 402040 KB Output is correct
12 Correct 257 ms 402040 KB Output is correct
13 Correct 263 ms 401912 KB Output is correct
14 Correct 259 ms 401916 KB Output is correct
15 Correct 257 ms 401888 KB Output is correct
16 Correct 262 ms 402040 KB Output is correct
17 Correct 260 ms 401968 KB Output is correct
18 Correct 264 ms 402040 KB Output is correct
19 Correct 259 ms 402020 KB Output is correct
20 Correct 263 ms 402012 KB Output is correct
21 Correct 257 ms 401912 KB Output is correct

Subtask #2
7.0 / 7.0

# Verdict Execution time Memory Grader output
1 Correct 256 ms 401912 KB Output is correct
2 Correct 260 ms 402040 KB Output is correct
3 Correct 260 ms 402048 KB Output is correct
4 Correct 261 ms 402040 KB Output is correct
5 Correct 259 ms 401912 KB Output is correct
6 Correct 258 ms 402040 KB Output is correct
7 Correct 259 ms 401912 KB Output is correct
8 Correct 259 ms 401912 KB Output is correct
9 Correct 262 ms 401912 KB Output is correct
10 Correct 262 ms 401912 KB Output is correct
11 Correct 259 ms 402040 KB Output is correct
12 Correct 257 ms 402040 KB Output is correct
13 Correct 263 ms 401912 KB Output is correct
14 Correct 259 ms 401916 KB Output is correct
15 Correct 257 ms 401888 KB Output is correct
16 Correct 262 ms 402040 KB Output is correct
17 Correct 259 ms 402424 KB Output is correct
18 Correct 261 ms 402424 KB Output is correct
19 Correct 257 ms 402552 KB Output is correct
20 Correct 261 ms 402040 KB Output is correct
21 Correct 260 ms 402296 KB Output is correct
22 Correct 259 ms 402296 KB Output is correct
23 Correct 261 ms 402424 KB Output is correct
24 Correct 258 ms 402040 KB Output is correct
25 Correct 260 ms 401968 KB Output is correct
26 Correct 264 ms 402040 KB Output is correct
27 Correct 259 ms 402020 KB Output is correct
28 Correct 263 ms 402012 KB Output is correct
29 Correct 257 ms 401912 KB Output is correct

Subtask #3
12.0 / 12.0

# Verdict Execution time Memory Grader output
1 Correct 256 ms 401912 KB Output is correct
2 Correct 260 ms 402040 KB Output is correct
3 Correct 260 ms 402048 KB Output is correct
4 Correct 261 ms 402040 KB Output is correct
5 Correct 259 ms 401912 KB Output is correct
6 Correct 258 ms 402040 KB Output is correct
7 Correct 259 ms 401912 KB Output is correct
8 Correct 259 ms 401912 KB Output is correct
9 Correct 262 ms 401912 KB Output is correct
10 Correct 262 ms 401912 KB Output is correct
11 Correct 259 ms 402040 KB Output is correct
12 Correct 257 ms 402040 KB Output is correct
13 Correct 263 ms 401912 KB Output is correct
14 Correct 259 ms 401916 KB Output is correct
15 Correct 257 ms 401888 KB Output is correct
16 Correct 262 ms 402040 KB Output is correct
17 Correct 259 ms 402424 KB Output is correct
18 Correct 261 ms 402424 KB Output is correct
19 Correct 257 ms 402552 KB Output is correct
20 Correct 261 ms 402040 KB Output is correct
21 Correct 260 ms 402296 KB Output is correct
22 Correct 259 ms 402296 KB Output is correct
23 Correct 261 ms 402424 KB Output is correct
24 Correct 258 ms 402040 KB Output is correct
25 Correct 266 ms 404916 KB Output is correct
26 Correct 268 ms 404852 KB Output is correct
27 Correct 268 ms 404852 KB Output is correct
28 Correct 263 ms 403064 KB Output is correct
29 Correct 269 ms 404344 KB Output is correct
30 Correct 271 ms 404344 KB Output is correct
31 Correct 268 ms 404216 KB Output is correct
32 Correct 269 ms 404088 KB Output is correct
33 Correct 260 ms 401968 KB Output is correct
34 Correct 264 ms 402040 KB Output is correct
35 Correct 259 ms 402020 KB Output is correct
36 Correct 263 ms 402012 KB Output is correct
37 Correct 257 ms 401912 KB Output is correct

Subtask #4
22.0 / 22.0

# Verdict Execution time Memory Grader output
1 Correct 256 ms 401912 KB Output is correct
2 Correct 260 ms 402040 KB Output is correct
3 Correct 260 ms 402048 KB Output is correct
4 Correct 261 ms 402040 KB Output is correct
5 Correct 259 ms 401912 KB Output is correct
6 Correct 258 ms 402040 KB Output is correct
7 Correct 259 ms 401912 KB Output is correct
8 Correct 259 ms 401912 KB Output is correct
9 Correct 262 ms 401912 KB Output is correct
10 Correct 262 ms 401912 KB Output is correct
11 Correct 259 ms 402040 KB Output is correct
12 Correct 257 ms 402040 KB Output is correct
13 Correct 263 ms 401912 KB Output is correct
14 Correct 259 ms 401916 KB Output is correct
15 Correct 257 ms 401888 KB Output is correct
16 Correct 262 ms 402040 KB Output is correct
17 Correct 259 ms 402424 KB Output is correct
18 Correct 261 ms 402424 KB Output is correct
19 Correct 257 ms 402552 KB Output is correct
20 Correct 261 ms 402040 KB Output is correct
21 Correct 260 ms 402296 KB Output is correct
22 Correct 259 ms 402296 KB Output is correct
23 Correct 261 ms 402424 KB Output is correct
24 Correct 258 ms 402040 KB Output is correct
25 Correct 266 ms 404916 KB Output is correct
26 Correct 268 ms 404852 KB Output is correct
27 Correct 268 ms 404852 KB Output is correct
28 Correct 263 ms 403064 KB Output is correct
29 Correct 269 ms 404344 KB Output is correct
30 Correct 271 ms 404344 KB Output is correct
31 Correct 268 ms 404216 KB Output is correct
32 Correct 269 ms 404088 KB Output is correct
33 Correct 333 ms 423160 KB Output is correct
34 Correct 336 ms 425976 KB Output is correct
35 Correct 326 ms 421752 KB Output is correct
36 Correct 329 ms 424572 KB Output is correct
37 Correct 382 ms 438364 KB Output is correct
38 Correct 382 ms 438412 KB Output is correct
39 Correct 381 ms 438492 KB Output is correct
40 Correct 375 ms 436188 KB Output is correct
41 Correct 303 ms 409464 KB Output is correct
42 Correct 323 ms 414456 KB Output is correct
43 Correct 407 ms 431612 KB Output is correct
44 Correct 415 ms 431992 KB Output is correct
45 Correct 340 ms 416760 KB Output is correct
46 Correct 337 ms 416892 KB Output is correct
47 Correct 401 ms 429944 KB Output is correct
48 Correct 411 ms 430200 KB Output is correct
49 Correct 260 ms 401968 KB Output is correct
50 Correct 264 ms 402040 KB Output is correct
51 Correct 259 ms 402020 KB Output is correct
52 Correct 263 ms 402012 KB Output is correct
53 Correct 257 ms 401912 KB Output is correct

Subtask #5
10.0 / 10.0

# Verdict Execution time Memory Grader output
1 Correct 259 ms 402984 KB Output is correct
2 Correct 261 ms 402680 KB Output is correct
3 Correct 258 ms 402168 KB Output is correct
4 Correct 263 ms 401900 KB Output is correct
5 Correct 262 ms 402524 KB Output is correct
6 Correct 262 ms 402552 KB Output is correct
7 Correct 262 ms 402424 KB Output is correct
8 Correct 258 ms 402424 KB Output is correct
9 Correct 260 ms 402424 KB Output is correct
10 Correct 258 ms 402180 KB Output is correct
11 Correct 259 ms 402236 KB Output is correct

Subtask #6
13.0 / 13.0

# Verdict Execution time Memory Grader output
1 Correct 257 ms 401912 KB Output is correct
2 Correct 563 ms 447992 KB Output is correct
3 Correct 941 ms 499576 KB Output is correct
4 Correct 975 ms 499960 KB Output is correct
5 Correct 937 ms 500088 KB Output is correct
6 Correct 387 ms 426232 KB Output is correct
7 Correct 539 ms 448120 KB Output is correct
8 Correct 559 ms 451064 KB Output is correct
9 Correct 260 ms 401968 KB Output is correct
10 Correct 264 ms 402040 KB Output is correct
11 Correct 259 ms 402020 KB Output is correct
12 Correct 263 ms 402012 KB Output is correct
13 Correct 257 ms 401912 KB Output is correct

Subtask #7
28.0 / 28.0

# Verdict Execution time Memory Grader output
1 Correct 256 ms 401912 KB Output is correct
2 Correct 260 ms 402040 KB Output is correct
3 Correct 260 ms 402048 KB Output is correct
4 Correct 261 ms 402040 KB Output is correct
5 Correct 259 ms 401912 KB Output is correct
6 Correct 258 ms 402040 KB Output is correct
7 Correct 259 ms 401912 KB Output is correct
8 Correct 259 ms 401912 KB Output is correct
9 Correct 262 ms 401912 KB Output is correct
10 Correct 262 ms 401912 KB Output is correct
11 Correct 259 ms 402040 KB Output is correct
12 Correct 257 ms 402040 KB Output is correct
13 Correct 263 ms 401912 KB Output is correct
14 Correct 259 ms 401916 KB Output is correct
15 Correct 257 ms 401888 KB Output is correct
16 Correct 262 ms 402040 KB Output is correct
17 Correct 259 ms 402424 KB Output is correct
18 Correct 261 ms 402424 KB Output is correct
19 Correct 257 ms 402552 KB Output is correct
20 Correct 261 ms 402040 KB Output is correct
21 Correct 260 ms 402296 KB Output is correct
22 Correct 259 ms 402296 KB Output is correct
23 Correct 261 ms 402424 KB Output is correct
24 Correct 258 ms 402040 KB Output is correct
25 Correct 266 ms 404916 KB Output is correct
26 Correct 268 ms 404852 KB Output is correct
27 Correct 268 ms 404852 KB Output is correct
28 Correct 263 ms 403064 KB Output is correct
29 Correct 269 ms 404344 KB Output is correct
30 Correct 271 ms 404344 KB Output is correct
31 Correct 268 ms 404216 KB Output is correct
32 Correct 269 ms 404088 KB Output is correct
33 Correct 333 ms 423160 KB Output is correct
34 Correct 336 ms 425976 KB Output is correct
35 Correct 326 ms 421752 KB Output is correct
36 Correct 329 ms 424572 KB Output is correct
37 Correct 382 ms 438364 KB Output is correct
38 Correct 382 ms 438412 KB Output is correct
39 Correct 381 ms 438492 KB Output is correct
40 Correct 375 ms 436188 KB Output is correct
41 Correct 303 ms 409464 KB Output is correct
42 Correct 323 ms 414456 KB Output is correct
43 Correct 407 ms 431612 KB Output is correct
44 Correct 415 ms 431992 KB Output is correct
45 Correct 340 ms 416760 KB Output is correct
46 Correct 337 ms 416892 KB Output is correct
47 Correct 401 ms 429944 KB Output is correct
48 Correct 411 ms 430200 KB Output is correct
49 Correct 259 ms 402984 KB Output is correct
50 Correct 261 ms 402680 KB Output is correct
51 Correct 258 ms 402168 KB Output is correct
52 Correct 263 ms 401900 KB Output is correct
53 Correct 262 ms 402524 KB Output is correct
54 Correct 262 ms 402552 KB Output is correct
55 Correct 262 ms 402424 KB Output is correct
56 Correct 258 ms 402424 KB Output is correct
57 Correct 260 ms 402424 KB Output is correct
58 Correct 258 ms 402180 KB Output is correct
59 Correct 259 ms 402236 KB Output is correct
60 Correct 257 ms 401912 KB Output is correct
61 Correct 563 ms 447992 KB Output is correct
62 Correct 941 ms 499576 KB Output is correct
63 Correct 975 ms 499960 KB Output is correct
64 Correct 937 ms 500088 KB Output is correct
65 Correct 387 ms 426232 KB Output is correct
66 Correct 539 ms 448120 KB Output is correct
67 Correct 559 ms 451064 KB Output is correct
68 Correct 1417 ms 662648 KB Output is correct
69 Correct 1330 ms 694008 KB Output is correct
70 Correct 1301 ms 647500 KB Output is correct
71 Correct 1330 ms 679592 KB Output is correct
72 Correct 1966 ms 863948 KB Output is correct
73 Correct 1492 ms 626636 KB Output is correct
74 Correct 1586 ms 635996 KB Output is correct
75 Correct 2345 ms 776560 KB Output is correct
76 Correct 1559 ms 630008 KB Output is correct
77 Correct 1983 ms 686448 KB Output is correct
78 Correct 2454 ms 781560 KB Output is correct
79 Correct 1468 ms 612124 KB Output is correct
80 Correct 2350 ms 755132 KB Output is correct
81 Correct 2294 ms 743992 KB Output is correct
82 Correct 1262 ms 678420 KB Output is correct
83 Correct 1980 ms 863660 KB Output is correct
84 Correct 1976 ms 863872 KB Output is correct
85 Correct 1974 ms 863932 KB Output is correct
86 Correct 260 ms 401968 KB Output is correct
87 Correct 264 ms 402040 KB Output is correct
88 Correct 259 ms 402020 KB Output is correct
89 Correct 263 ms 402012 KB Output is correct
90 Correct 257 ms 401912 KB Output is correct