#include "rect.h"
#include <bits/stdc++.h>
using namespace std;
typedef long long ll;
typedef pair<int, int> pii;
typedef pair<ll, ll> pll;
const int MAXN = 2500;
int N, M, A[MAXN+10][MAXN+10];
int L[MAXN+10][MAXN+10], R[MAXN+10][MAXN+10], U[MAXN+10][MAXN+10], D[MAXN+10][MAXN+10];
struct Line
{
int y, x1, x2;
bool operator < (const Line &p) const
{
if(y!=p.y) return y<p.y;
if(x1!=p.x1) return x1<p.x1;
return x2<p.x2;
}
bool operator == (const Line &p) { return y==p.y && x1==p.x1 && x2==p.x2; }
bool operator != (const Line &p) { return !(y==p.y && x1==p.x1 && x2==p.x2); }
};
vector<Line> H, V;
int lowH[MAXN*MAXN+10], lowV[MAXN*MAXN+10];
ll count_rectangles(vector<vector<int>> _A)
{
int i, j;
N=_A.size(); M=_A[0].size();
for(i=1; i<=N; i++) for(j=1; j<=M; j++) A[i][j]=_A[i-1][j-1];
for(i=1; i<=N; i++)
{
vector<int> S;
S.clear();
for(j=1; j<=M; j++) L[i][j]=0;
for(j=1; j<=M; j++)
{
while(!S.empty() && A[i][S.back()]<=A[i][j]) S.pop_back();
if(!S.empty()) L[i][j]=S.back();
S.push_back(j);
}
S.clear();
for(j=1; j<=M; j++) R[i][j]=M+1;
for(j=M; j>=1; j--)
{
while(!S.empty() && A[i][S.back()]<=A[i][j]) S.pop_back();
if(!S.empty()) R[i][j]=S.back();
S.push_back(j);
}
}
for(i=1; i<=M; i++)
{
vector<int> S;
S.clear();
for(j=1; j<=N; j++) U[j][i]=0;
for(j=1; j<=N; j++)
{
while(!S.empty() && A[S.back()][i]<=A[j][i]) S.pop_back();
if(!S.empty()) U[j][i]=S.back();
S.push_back(j);
}
S.clear();
for(j=1; j<=N; j++) D[j][i]=N+1;
for(j=N; j>=1; j--)
{
while(!S.empty() && A[S.back()][i]<=A[j][i]) S.pop_back();
if(!S.empty()) D[j][i]=S.back();
S.push_back(j);
}
}
for(i=1; i<=N; i++)
{
for(j=1; j<=M; j++)
{
if(L[i][j]==0) continue;
if(R[i][j]==M+1) continue;
H.push_back({i, L[i][j]+1, R[i][j]-1});
}
for(j=1; j<=M; j++)
{
if(U[i][j]==0) continue;
if(D[i][j]==N+1) continue;
V.push_back({j, U[i][j]+1, D[i][j]-1});
}
}
sort(H.begin(), H.end());
H.erase(unique(H.begin(), H.end()), H.end());
sort(V.begin(), V.end());
V.erase(unique(V.begin(), V.end()), V.end());
for(i=H.size()-1; i>=0; i--)
{
lowH[i]=H[i].y;
Line t={H[i].y+1, H[i].x1, H[i].x2};
auto it=lower_bound(H.begin(), H.end(), t);
if(it!=H.end() && *it==t) lowH[i]=lowH[it-H.begin()];
}
for(i=V.size()-1; i>=0; i--)
{
lowV[i]=V[i].y;
Line t={V[i].y+1, V[i].x1, V[i].x2};
auto it=lower_bound(V.begin(), V.end(), t);
if(it!=V.end() && *it==t) lowV[i]=lowV[it-V.begin()];
}
vector<pair<pii, pii>> ans;
for(i=1; i<=N; i++) for(j=1; j<=M; j++)
{
//printf("%d %d : %d %d %d %d\n", i, j, L[i][j], R[i][j], U[i][j], D[i][j]);
if(L[i][j]==0) continue;
if(R[i][j]==M+1) continue;
if(U[i][j]==0) continue;
if(D[i][j]==N+1) continue;
Line tu={U[i][j]+1, L[i][j]+1, R[i][j]-1};
Line tl={L[i][j]+1, U[i][j]+1, D[i][j]-1};
auto it=lower_bound(H.begin(), H.end(), tu);
auto jt=lower_bound(V.begin(), V.end(), tl);
if(it!=H.end() && *it!=tu) continue;
if(jt!=V.end() && *jt!=tl) continue;
if(lowH[it-H.begin()]<D[i][j]-1) continue;
if(lowV[jt-V.begin()]<R[i][j]-1) continue;
ans.push_back({pii(L[i][j]+1, R[i][j]-1), pii(U[i][j]+1, D[i][j]-1)});
}
sort(ans.begin(), ans.end());
//ans.erase(unique(ans.begin(), ans.end()), ans.end());
return ans.size();
}
# |
Verdict |
Execution time |
Memory |
Grader output |
1 |
Correct |
2 ms |
504 KB |
Output is correct |
2 |
Correct |
3 ms |
1016 KB |
Output is correct |
3 |
Correct |
3 ms |
1016 KB |
Output is correct |
4 |
Correct |
3 ms |
1016 KB |
Output is correct |
5 |
Correct |
3 ms |
1016 KB |
Output is correct |
6 |
Incorrect |
3 ms |
1016 KB |
Output isn't correct |
7 |
Halted |
0 ms |
0 KB |
- |
# |
Verdict |
Execution time |
Memory |
Grader output |
1 |
Correct |
2 ms |
504 KB |
Output is correct |
2 |
Correct |
3 ms |
1016 KB |
Output is correct |
3 |
Correct |
3 ms |
1016 KB |
Output is correct |
4 |
Correct |
3 ms |
1016 KB |
Output is correct |
5 |
Correct |
3 ms |
1016 KB |
Output is correct |
6 |
Incorrect |
3 ms |
1016 KB |
Output isn't correct |
7 |
Halted |
0 ms |
0 KB |
- |
# |
Verdict |
Execution time |
Memory |
Grader output |
1 |
Correct |
2 ms |
504 KB |
Output is correct |
2 |
Correct |
3 ms |
1016 KB |
Output is correct |
3 |
Correct |
3 ms |
1016 KB |
Output is correct |
4 |
Correct |
3 ms |
1016 KB |
Output is correct |
5 |
Correct |
3 ms |
1016 KB |
Output is correct |
6 |
Incorrect |
3 ms |
1016 KB |
Output isn't correct |
7 |
Halted |
0 ms |
0 KB |
- |
# |
Verdict |
Execution time |
Memory |
Grader output |
1 |
Correct |
2 ms |
504 KB |
Output is correct |
2 |
Correct |
3 ms |
1016 KB |
Output is correct |
3 |
Correct |
3 ms |
1016 KB |
Output is correct |
4 |
Correct |
3 ms |
1016 KB |
Output is correct |
5 |
Correct |
3 ms |
1016 KB |
Output is correct |
6 |
Incorrect |
3 ms |
1016 KB |
Output isn't correct |
7 |
Halted |
0 ms |
0 KB |
- |
# |
Verdict |
Execution time |
Memory |
Grader output |
1 |
Correct |
5 ms |
888 KB |
Output is correct |
2 |
Correct |
2 ms |
888 KB |
Output is correct |
3 |
Correct |
3 ms |
632 KB |
Output is correct |
4 |
Correct |
2 ms |
376 KB |
Output is correct |
5 |
Correct |
4 ms |
888 KB |
Output is correct |
6 |
Correct |
4 ms |
888 KB |
Output is correct |
7 |
Correct |
5 ms |
888 KB |
Output is correct |
8 |
Correct |
5 ms |
888 KB |
Output is correct |
9 |
Correct |
4 ms |
888 KB |
Output is correct |
10 |
Correct |
2 ms |
504 KB |
Output is correct |
11 |
Correct |
4 ms |
760 KB |
Output is correct |
# |
Verdict |
Execution time |
Memory |
Grader output |
1 |
Correct |
2 ms |
504 KB |
Output is correct |
2 |
Incorrect |
1343 ms |
130000 KB |
Output isn't correct |
3 |
Halted |
0 ms |
0 KB |
- |
# |
Verdict |
Execution time |
Memory |
Grader output |
1 |
Correct |
2 ms |
504 KB |
Output is correct |
2 |
Correct |
3 ms |
1016 KB |
Output is correct |
3 |
Correct |
3 ms |
1016 KB |
Output is correct |
4 |
Correct |
3 ms |
1016 KB |
Output is correct |
5 |
Correct |
3 ms |
1016 KB |
Output is correct |
6 |
Incorrect |
3 ms |
1016 KB |
Output isn't correct |
7 |
Halted |
0 ms |
0 KB |
- |