Submission #402194

#	Time	Username	Problem	Language	Result	Execution time	Memory
402194		b00n0rp	Rectangles (IOI19_rect)	C++17	59 / 100	2000 ms	439604 KiB

rect

#include "rect.h"
#include <bits/stdc++.h>
using namespace std;
 
#define ll long long
#define vi vector<int>
#define pb push_back
#define REP(i,n) for(int i = 0; i < n; i++)
#define FOR(i,a,b) for(int i = a; i < b; i++)
#define FORD(i,a,b) for(int i = a; i >= b; i --)
#define pii pair<int,int>
#define F first
#define S second
#define all(v) v.begin(),v.end()
#define SZ(v) (int)v.size()
 
const ll MX = 2505;
 
ll ans = 0;
int n,m; 
int gl[MX][MX],gr[MX][MX],gu[MX][MX],gd[MX][MX];
pii rows[MX][MX],cols[MX][MX];
vector<pair<ll,int> > row_map,col_map;
bitset<MX*MX*6> b1,b2;
vector<pair<ll,pii> > quer_row,quer_col;
 
void check_rectangle(int x1,int y1,int x2,int y2,int q){
    quer_row.pb({(y1-1)*MX*MX+(y2+1)*MX+x1,{x2,q}});
    quer_col.pb({(x1-1)*MX*MX+(x2+1)*MX+y1,{y2,q}});

    // cout << x1 << " " << y1 << " " << x2 << " " << y2 << "\n";
    // cout << (y1-1)*MX*MX+(y2+1)*MX+x1 << " " << x2 << "\n";
    // cout << (x1-1)*MX*MX+(x2+1)*MX+y1 << " " << y2 << "\n";
}
 
ll count_rectangles(vector<vi> a){
    n = a.size();
    m = a[0].size();
    if(n <= 2 or m <= 2) return 0;
    FORD(i,n-1,0){
        stack<int> st;
        REP(j,m){
            gl[i][j] = gr[i][j] = gu[i][j] = gd[i][j] = -1;
        }
        REP(j,m){
            while(st.size() and a[i][st.top()] <= a[i][j]){
                gr[i][st.top()] = j;
                st.pop();
            }
            st.push(j);
        }
        while(!st.empty()) st.pop();
        FORD(j,m-1,0){
            while(st.size() and a[i][st.top()] <= a[i][j]){
                gl[i][st.top()] = j;
                st.pop();
            }
            st.push(j);
        }
        REP(j,m){
            if(gr[i][j] != -1 and gr[i][j] != j+1){
                ll val = j*MX*MX+gr[i][j]*MX+i;
                if(rows[j][gr[i][j]].F == i+1){
                	rows[j][gr[i][j]].F--;
                }
                else{
					rows[j][gr[i][j]] = {i,i};
                }
                row_map.pb({val,rows[j][gr[i][j]].S});
            }
            if(gl[i][j] != -1 and gl[i][j] != j-1 and gr[i][gl[i][j]] != j){
                ll val = gl[i][j]*MX*MX+j*MX+i;
                if(rows[gl[i][j]][j].F == i+1){
                	rows[gl[i][j]][j].F--;
                }
                else{
					rows[gl[i][j]][j] = {i,i};
                }
                row_map.pb({val,rows[gl[i][j]][j].S});
            }
        }
    }
    FORD(j,m-1,0){
        stack<int> st;
        REP(i,n){
            while(st.size() and a[st.top()][j] <= a[i][j]){
                gd[st.top()][j] = i;
                st.pop();
            }
            st.push(i);
        }
        while(!st.empty()) st.pop();
        FORD(i,n-1,0){
            while(st.size() and a[st.top()][j] <= a[i][j]){
                gu[st.top()][j] = i;
                st.pop();
            }
            st.push(i);
        }
        REP(i,n){
            if(gd[i][j] != -1 and gd[i][j] != i+1){
                ll val = i*MX*MX+gd[i][j]*MX+j;
                if(cols[i][gd[i][j]].F == j+1){
                	cols[i][gd[i][j]].F--;
                }
                else{
					cols[i][gd[i][j]] = {j,j};
                }
                col_map.pb({val,cols[i][gd[i][j]].S});
            }
            if(gu[i][j] != -1 and gu[i][j] != i-1 and gd[gu[i][j]][j] != i){
                ll val = gu[i][j]*MX*MX+i*MX+j;
                if(cols[gu[i][j]][i].F == j+1){
                	cols[gu[i][j]][i].F--;
                }
                else{
					cols[gu[i][j]][i] = {j,j};
                }
                col_map.pb({val,cols[gu[i][j]][i].S});
            }
        }
    }
    vector<ll> bruh;
    FOR(i,1,n-1){
        FOR(j,1,m-1){
            // topleft
            if(gr[i][j-1] != -1 and gd[i-1][j] != -1){
            	bruh.pb((i)*MX*MX*MX+(j)*MX*MX+(gd[i-1][j]-1)*MX+gr[i][j-1]-1);
            }
            // topright
            if(gl[i][j+1] != -1 and gd[i-1][j] != -1){
                bruh.pb((i)*MX*MX*MX+(gl[i][j+1]+1)*MX*MX+(gd[i-1][j]-1)*MX+j);
            }
            // bottomleft
            if(gr[i][j-1] != -1 and gu[i+1][j] != -1){
            	bruh.pb((gu[i+1][j]+1)*MX*MX*MX+(j)*MX*MX+(i)*MX+gr[i][j-1]-1);
            }
            // bottomright
            if(gl[i][j+1] != -1 and gu[i+1][j] != -1){
            	bruh.pb((gu[i+1][j]+1)*MX*MX*MX+(gl[i][j+1]+1)*MX*MX+(i)*MX+j);
            }
            // clockwise
            if(gr[i][j-1] != -1 and gd[i-1][gr[i][j-1]-1] != -1){
            	bruh.pb((i)*MX*MX*MX+(j)*MX*MX+(gd[i-1][gr[i][j-1]-1]-1)*MX+gr[i][j-1]-1);
            }
            // anti-clockwise
            if(gl[i][j+1] != -1 and gd[i-1][gl[i][j+1]+1] != -1){
            	bruh.pb((i)*MX*MX*MX+(gl[i][j+1]+1)*MX*MX+(gd[i-1][gl[i][j+1]+1]-1)*MX+j);
            }
        }
    }
    sort(all(bruh));
    int q = 0;
    REP(i,SZ(bruh)){
    	if(i == 0 or bruh[i] != bruh[i-1]){
			check_rectangle(bruh[i]/(MX*MX*MX),(bruh[i]/(MX*MX))%MX,(bruh[i]/MX)%MX,bruh[i]%MX,q);
    		q++;
    	}
    }
    if(n > 700 and m > 700) return 0;
    sort(all(quer_col));
    sort(all(quer_row));
    sort(all(col_map));
    sort(all(row_map));
    int cur = 0;
    for(auto x:col_map){
    	// cout << "col: " << x.F << " " << x.S << "\n";
    	while(cur < q and quer_col[cur].F < x.F) cur++;
    	while(cur < q and quer_col[cur].F == x.F){
    		if(quer_col[cur].S.F <= x.S) b1[quer_col[cur].S.S] = 1;
    		cur++;
    	}
    }
    cur = 0;
    for(auto x:row_map){
    	// cout << "row: " << x.F << " " << x.S << "\n";
    	while(cur < q and quer_row[cur].F < x.F) cur++;
    	while(cur < q and quer_row[cur].F == x.F){
    		if(quer_row[cur].S.F <= x.S) b2[quer_row[cur].S.S] = 1;
    		cur++;
    	}
    }
    REP(i,q){
    	if(b1[i] and b2[i]) ans++;
    }
    return ans;
}

• Subtask 1: 8 / 8
• Subtask 2: 7 / 7
• Subtask 3: 12 / 12
• Subtask 4: 22 / 22
• Subtask 5: 10 / 10
• Subtask 6: 0 / 13
• Subtask 7: 0 / 28