#pragma GCC optimize ("O3")
#pragma GCC optimize ("Ofast")
#pragma GCC optimize ("unroll-loops")
#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];
vector<pair<pii, pii>> ans;
int posH[MAXN+10][MAXN+10], posV[MAXN+10][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, j=H.size()-1; i>=0; i--)
{
lowH[i]=H[i].y;
Line t={H[i].y+1, H[i].x1, H[i].x2};
for(; j>=0 && t<H[j]; j--);
if(j>=0 && t==H[j]) lowH[i]=lowH[j];
}
for(i=V.size()-1, j=V.size()-1; i>=0; i--)
{
lowV[i]=V[i].y;
Line t={V[i].y+1, V[i].x1, V[i].x2};
for(; j>=0 && t<V[j]; j--);
if(j>=0 && t==V[j]) lowV[i]=lowV[j];
}
vector<pair<Line, pii>> HT, VT;
for(i=2; i<=N-1; i++) for(j=2; j<=M-1; 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};
HT.push_back({tu, pii(i, j)});
VT.push_back({tl, pii(i, j)});
}
sort(HT.begin(), HT.end());
for(i=0, j=0; i<HT.size(); i++)
{
for(; j<H.size() && H[j]<HT[i].first; j++);
if(j<H.size() && HT[i].first==H[j]) posH[HT[i].second.first][HT[i].second.second]=j;
else posH[HT[i].second.first][HT[i].second.second]=-1;
}
sort(VT.begin(), VT.end());
for(i=0, j=0; i<VT.size(); i++)
{
for(; j<V.size() && V[j]<VT[i].first; j++);
if(j<V.size() && VT[i].first==V[j]) posV[VT[i].second.first][VT[i].second.second]=j;
else posV[VT[i].second.first][VT[i].second.second]=-1;
}
for(i=2; i<=N-1; i++) for(j=2; j<=M-1; 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;
int it=posH[i][j];
int jt=posV[i][j];
if(it==-1 || jt==-1) continue;
if(lowH[it]<D[i][j]-1) continue;
if(lowV[jt]<R[i][j]-1) continue;
ans.push_back({pii(L[i][j], R[i][j]), pii(U[i][j], D[i][j])});
}
sort(ans.begin(), ans.end());
ans.erase(unique(ans.begin(), ans.end()), ans.end());
return ans.size();
}
Compilation message
rect.cpp: In function 'll count_rectangles(std::vector<std::vector<int> >)':
rect.cpp:142:18: warning: comparison between signed and unsigned integer expressions [-Wsign-compare]
for(i=0, j=0; i<HT.size(); i++)
~^~~~~~~~~~
rect.cpp:144:11: warning: comparison between signed and unsigned integer expressions [-Wsign-compare]
for(; j<H.size() && H[j]<HT[i].first; j++);
~^~~~~~~~~
rect.cpp:145:8: warning: comparison between signed and unsigned integer expressions [-Wsign-compare]
if(j<H.size() && HT[i].first==H[j]) posH[HT[i].second.first][HT[i].second.second]=j;
~^~~~~~~~~
rect.cpp:150:18: warning: comparison between signed and unsigned integer expressions [-Wsign-compare]
for(i=0, j=0; i<VT.size(); i++)
~^~~~~~~~~~
rect.cpp:152:11: warning: comparison between signed and unsigned integer expressions [-Wsign-compare]
for(; j<V.size() && V[j]<VT[i].first; j++);
~^~~~~~~~~
rect.cpp:153:8: warning: comparison between signed and unsigned integer expressions [-Wsign-compare]
if(j<V.size() && VT[i].first==V[j]) posV[VT[i].second.first][VT[i].second.second]=j;
~^~~~~~~~~
# |
Verdict |
Execution time |
Memory |
Grader output |
1 |
Correct |
2 ms |
504 KB |
Output is correct |
2 |
Correct |
3 ms |
1272 KB |
Output is correct |
3 |
Correct |
3 ms |
1400 KB |
Output is correct |
4 |
Correct |
3 ms |
1528 KB |
Output is correct |
5 |
Correct |
3 ms |
1016 KB |
Output is correct |
6 |
Correct |
3 ms |
1272 KB |
Output is correct |
7 |
Correct |
3 ms |
1272 KB |
Output is correct |
8 |
Correct |
2 ms |
760 KB |
Output is correct |
9 |
Correct |
3 ms |
1272 KB |
Output is correct |
10 |
Correct |
3 ms |
1272 KB |
Output is correct |
11 |
Correct |
3 ms |
1272 KB |
Output is correct |
12 |
Correct |
3 ms |
1272 KB |
Output is correct |
13 |
Correct |
2 ms |
376 KB |
Output is correct |
14 |
Correct |
2 ms |
504 KB |
Output is correct |
15 |
Correct |
2 ms |
504 KB |
Output is correct |
16 |
Correct |
2 ms |
504 KB |
Output is correct |
17 |
Correct |
2 ms |
376 KB |
Output is correct |
18 |
Correct |
2 ms |
376 KB |
Output is correct |
19 |
Correct |
3 ms |
1272 KB |
Output is correct |
20 |
Correct |
2 ms |
888 KB |
Output is correct |
21 |
Correct |
2 ms |
504 KB |
Output is correct |
# |
Verdict |
Execution time |
Memory |
Grader output |
1 |
Correct |
2 ms |
504 KB |
Output is correct |
2 |
Correct |
3 ms |
1272 KB |
Output is correct |
3 |
Correct |
3 ms |
1400 KB |
Output is correct |
4 |
Correct |
3 ms |
1528 KB |
Output is correct |
5 |
Correct |
3 ms |
1016 KB |
Output is correct |
6 |
Correct |
3 ms |
1272 KB |
Output is correct |
7 |
Correct |
3 ms |
1272 KB |
Output is correct |
8 |
Correct |
2 ms |
760 KB |
Output is correct |
9 |
Correct |
3 ms |
1272 KB |
Output is correct |
10 |
Correct |
3 ms |
1272 KB |
Output is correct |
11 |
Correct |
3 ms |
1272 KB |
Output is correct |
12 |
Correct |
3 ms |
1272 KB |
Output is correct |
13 |
Correct |
2 ms |
376 KB |
Output is correct |
14 |
Correct |
2 ms |
504 KB |
Output is correct |
15 |
Correct |
2 ms |
504 KB |
Output is correct |
16 |
Correct |
2 ms |
504 KB |
Output is correct |
17 |
Correct |
8 ms |
3612 KB |
Output is correct |
18 |
Correct |
8 ms |
3612 KB |
Output is correct |
19 |
Correct |
8 ms |
3612 KB |
Output is correct |
20 |
Correct |
7 ms |
3192 KB |
Output is correct |
21 |
Correct |
8 ms |
3448 KB |
Output is correct |
22 |
Correct |
8 ms |
3476 KB |
Output is correct |
23 |
Correct |
8 ms |
3448 KB |
Output is correct |
24 |
Correct |
6 ms |
3064 KB |
Output is correct |
25 |
Correct |
2 ms |
376 KB |
Output is correct |
26 |
Correct |
2 ms |
376 KB |
Output is correct |
27 |
Correct |
3 ms |
1272 KB |
Output is correct |
28 |
Correct |
2 ms |
888 KB |
Output is correct |
29 |
Correct |
2 ms |
504 KB |
Output is correct |
# |
Verdict |
Execution time |
Memory |
Grader output |
1 |
Correct |
2 ms |
504 KB |
Output is correct |
2 |
Correct |
3 ms |
1272 KB |
Output is correct |
3 |
Correct |
3 ms |
1400 KB |
Output is correct |
4 |
Correct |
3 ms |
1528 KB |
Output is correct |
5 |
Correct |
3 ms |
1016 KB |
Output is correct |
6 |
Correct |
3 ms |
1272 KB |
Output is correct |
7 |
Correct |
3 ms |
1272 KB |
Output is correct |
8 |
Correct |
2 ms |
760 KB |
Output is correct |
9 |
Correct |
3 ms |
1272 KB |
Output is correct |
10 |
Correct |
3 ms |
1272 KB |
Output is correct |
11 |
Correct |
3 ms |
1272 KB |
Output is correct |
12 |
Correct |
3 ms |
1272 KB |
Output is correct |
13 |
Correct |
2 ms |
376 KB |
Output is correct |
14 |
Correct |
2 ms |
504 KB |
Output is correct |
15 |
Correct |
2 ms |
504 KB |
Output is correct |
16 |
Correct |
2 ms |
504 KB |
Output is correct |
17 |
Correct |
8 ms |
3612 KB |
Output is correct |
18 |
Correct |
8 ms |
3612 KB |
Output is correct |
19 |
Correct |
8 ms |
3612 KB |
Output is correct |
20 |
Correct |
7 ms |
3192 KB |
Output is correct |
21 |
Correct |
8 ms |
3448 KB |
Output is correct |
22 |
Correct |
8 ms |
3476 KB |
Output is correct |
23 |
Correct |
8 ms |
3448 KB |
Output is correct |
24 |
Correct |
6 ms |
3064 KB |
Output is correct |
25 |
Correct |
32 ms |
11872 KB |
Output is correct |
26 |
Correct |
32 ms |
11828 KB |
Output is correct |
27 |
Correct |
32 ms |
11876 KB |
Output is correct |
28 |
Correct |
27 ms |
9412 KB |
Output is correct |
29 |
Correct |
37 ms |
11112 KB |
Output is correct |
30 |
Correct |
38 ms |
11112 KB |
Output is correct |
31 |
Correct |
37 ms |
11048 KB |
Output is correct |
32 |
Correct |
36 ms |
10792 KB |
Output is correct |
33 |
Correct |
2 ms |
376 KB |
Output is correct |
34 |
Correct |
2 ms |
376 KB |
Output is correct |
35 |
Correct |
3 ms |
1272 KB |
Output is correct |
36 |
Correct |
2 ms |
888 KB |
Output is correct |
37 |
Correct |
2 ms |
504 KB |
Output is correct |
# |
Verdict |
Execution time |
Memory |
Grader output |
1 |
Correct |
2 ms |
504 KB |
Output is correct |
2 |
Correct |
3 ms |
1272 KB |
Output is correct |
3 |
Correct |
3 ms |
1400 KB |
Output is correct |
4 |
Correct |
3 ms |
1528 KB |
Output is correct |
5 |
Correct |
3 ms |
1016 KB |
Output is correct |
6 |
Correct |
3 ms |
1272 KB |
Output is correct |
7 |
Correct |
3 ms |
1272 KB |
Output is correct |
8 |
Correct |
2 ms |
760 KB |
Output is correct |
9 |
Correct |
3 ms |
1272 KB |
Output is correct |
10 |
Correct |
3 ms |
1272 KB |
Output is correct |
11 |
Correct |
3 ms |
1272 KB |
Output is correct |
12 |
Correct |
3 ms |
1272 KB |
Output is correct |
13 |
Correct |
2 ms |
376 KB |
Output is correct |
14 |
Correct |
2 ms |
504 KB |
Output is correct |
15 |
Correct |
2 ms |
504 KB |
Output is correct |
16 |
Correct |
2 ms |
504 KB |
Output is correct |
17 |
Correct |
8 ms |
3612 KB |
Output is correct |
18 |
Correct |
8 ms |
3612 KB |
Output is correct |
19 |
Correct |
8 ms |
3612 KB |
Output is correct |
20 |
Correct |
7 ms |
3192 KB |
Output is correct |
21 |
Correct |
8 ms |
3448 KB |
Output is correct |
22 |
Correct |
8 ms |
3476 KB |
Output is correct |
23 |
Correct |
8 ms |
3448 KB |
Output is correct |
24 |
Correct |
6 ms |
3064 KB |
Output is correct |
25 |
Correct |
32 ms |
11872 KB |
Output is correct |
26 |
Correct |
32 ms |
11828 KB |
Output is correct |
27 |
Correct |
32 ms |
11876 KB |
Output is correct |
28 |
Correct |
27 ms |
9412 KB |
Output is correct |
29 |
Correct |
37 ms |
11112 KB |
Output is correct |
30 |
Correct |
38 ms |
11112 KB |
Output is correct |
31 |
Correct |
37 ms |
11048 KB |
Output is correct |
32 |
Correct |
36 ms |
10792 KB |
Output is correct |
33 |
Correct |
125 ms |
41340 KB |
Output is correct |
34 |
Correct |
136 ms |
41344 KB |
Output is correct |
35 |
Correct |
127 ms |
41304 KB |
Output is correct |
36 |
Correct |
122 ms |
41332 KB |
Output is correct |
37 |
Correct |
379 ms |
87588 KB |
Output is correct |
38 |
Correct |
379 ms |
87452 KB |
Output is correct |
39 |
Correct |
379 ms |
87548 KB |
Output is correct |
40 |
Correct |
362 ms |
82420 KB |
Output is correct |
41 |
Correct |
225 ms |
57832 KB |
Output is correct |
42 |
Correct |
300 ms |
61136 KB |
Output is correct |
43 |
Correct |
439 ms |
77572 KB |
Output is correct |
44 |
Correct |
443 ms |
77740 KB |
Output is correct |
45 |
Correct |
223 ms |
49668 KB |
Output is correct |
46 |
Correct |
212 ms |
39688 KB |
Output is correct |
47 |
Correct |
451 ms |
78244 KB |
Output is correct |
48 |
Correct |
452 ms |
77884 KB |
Output is correct |
49 |
Correct |
2 ms |
376 KB |
Output is correct |
50 |
Correct |
2 ms |
376 KB |
Output is correct |
51 |
Correct |
3 ms |
1272 KB |
Output is correct |
52 |
Correct |
2 ms |
888 KB |
Output is correct |
53 |
Correct |
2 ms |
504 KB |
Output is correct |
# |
Verdict |
Execution time |
Memory |
Grader output |
1 |
Correct |
4 ms |
1144 KB |
Output is correct |
2 |
Correct |
4 ms |
1016 KB |
Output is correct |
3 |
Correct |
3 ms |
632 KB |
Output is correct |
4 |
Correct |
1 ms |
376 KB |
Output is correct |
5 |
Correct |
4 ms |
888 KB |
Output is correct |
6 |
Correct |
4 ms |
760 KB |
Output is correct |
7 |
Correct |
4 ms |
760 KB |
Output is correct |
8 |
Correct |
4 ms |
760 KB |
Output is correct |
9 |
Correct |
4 ms |
888 KB |
Output is correct |
10 |
Correct |
3 ms |
504 KB |
Output is correct |
11 |
Correct |
3 ms |
760 KB |
Output is correct |
# |
Verdict |
Execution time |
Memory |
Grader output |
1 |
Correct |
2 ms |
632 KB |
Output is correct |
2 |
Correct |
1388 ms |
209212 KB |
Output is correct |
3 |
Correct |
3118 ms |
439864 KB |
Output is correct |
4 |
Correct |
3154 ms |
442128 KB |
Output is correct |
5 |
Correct |
3151 ms |
442120 KB |
Output is correct |
6 |
Correct |
307 ms |
85240 KB |
Output is correct |
7 |
Correct |
675 ms |
169264 KB |
Output is correct |
8 |
Correct |
721 ms |
172448 KB |
Output is correct |
9 |
Correct |
2 ms |
376 KB |
Output is correct |
10 |
Correct |
2 ms |
376 KB |
Output is correct |
11 |
Correct |
3 ms |
1272 KB |
Output is correct |
12 |
Correct |
2 ms |
888 KB |
Output is correct |
13 |
Correct |
2 ms |
504 KB |
Output is correct |
# |
Verdict |
Execution time |
Memory |
Grader output |
1 |
Correct |
2 ms |
504 KB |
Output is correct |
2 |
Correct |
3 ms |
1272 KB |
Output is correct |
3 |
Correct |
3 ms |
1400 KB |
Output is correct |
4 |
Correct |
3 ms |
1528 KB |
Output is correct |
5 |
Correct |
3 ms |
1016 KB |
Output is correct |
6 |
Correct |
3 ms |
1272 KB |
Output is correct |
7 |
Correct |
3 ms |
1272 KB |
Output is correct |
8 |
Correct |
2 ms |
760 KB |
Output is correct |
9 |
Correct |
3 ms |
1272 KB |
Output is correct |
10 |
Correct |
3 ms |
1272 KB |
Output is correct |
11 |
Correct |
3 ms |
1272 KB |
Output is correct |
12 |
Correct |
3 ms |
1272 KB |
Output is correct |
13 |
Correct |
2 ms |
376 KB |
Output is correct |
14 |
Correct |
2 ms |
504 KB |
Output is correct |
15 |
Correct |
2 ms |
504 KB |
Output is correct |
16 |
Correct |
2 ms |
504 KB |
Output is correct |
17 |
Correct |
8 ms |
3612 KB |
Output is correct |
18 |
Correct |
8 ms |
3612 KB |
Output is correct |
19 |
Correct |
8 ms |
3612 KB |
Output is correct |
20 |
Correct |
7 ms |
3192 KB |
Output is correct |
21 |
Correct |
8 ms |
3448 KB |
Output is correct |
22 |
Correct |
8 ms |
3476 KB |
Output is correct |
23 |
Correct |
8 ms |
3448 KB |
Output is correct |
24 |
Correct |
6 ms |
3064 KB |
Output is correct |
25 |
Correct |
32 ms |
11872 KB |
Output is correct |
26 |
Correct |
32 ms |
11828 KB |
Output is correct |
27 |
Correct |
32 ms |
11876 KB |
Output is correct |
28 |
Correct |
27 ms |
9412 KB |
Output is correct |
29 |
Correct |
37 ms |
11112 KB |
Output is correct |
30 |
Correct |
38 ms |
11112 KB |
Output is correct |
31 |
Correct |
37 ms |
11048 KB |
Output is correct |
32 |
Correct |
36 ms |
10792 KB |
Output is correct |
33 |
Correct |
125 ms |
41340 KB |
Output is correct |
34 |
Correct |
136 ms |
41344 KB |
Output is correct |
35 |
Correct |
127 ms |
41304 KB |
Output is correct |
36 |
Correct |
122 ms |
41332 KB |
Output is correct |
37 |
Correct |
379 ms |
87588 KB |
Output is correct |
38 |
Correct |
379 ms |
87452 KB |
Output is correct |
39 |
Correct |
379 ms |
87548 KB |
Output is correct |
40 |
Correct |
362 ms |
82420 KB |
Output is correct |
41 |
Correct |
225 ms |
57832 KB |
Output is correct |
42 |
Correct |
300 ms |
61136 KB |
Output is correct |
43 |
Correct |
439 ms |
77572 KB |
Output is correct |
44 |
Correct |
443 ms |
77740 KB |
Output is correct |
45 |
Correct |
223 ms |
49668 KB |
Output is correct |
46 |
Correct |
212 ms |
39688 KB |
Output is correct |
47 |
Correct |
451 ms |
78244 KB |
Output is correct |
48 |
Correct |
452 ms |
77884 KB |
Output is correct |
49 |
Correct |
4 ms |
1144 KB |
Output is correct |
50 |
Correct |
4 ms |
1016 KB |
Output is correct |
51 |
Correct |
3 ms |
632 KB |
Output is correct |
52 |
Correct |
1 ms |
376 KB |
Output is correct |
53 |
Correct |
4 ms |
888 KB |
Output is correct |
54 |
Correct |
4 ms |
760 KB |
Output is correct |
55 |
Correct |
4 ms |
760 KB |
Output is correct |
56 |
Correct |
4 ms |
760 KB |
Output is correct |
57 |
Correct |
4 ms |
888 KB |
Output is correct |
58 |
Correct |
3 ms |
504 KB |
Output is correct |
59 |
Correct |
3 ms |
760 KB |
Output is correct |
60 |
Correct |
2 ms |
632 KB |
Output is correct |
61 |
Correct |
1388 ms |
209212 KB |
Output is correct |
62 |
Correct |
3118 ms |
439864 KB |
Output is correct |
63 |
Correct |
3154 ms |
442128 KB |
Output is correct |
64 |
Correct |
3151 ms |
442120 KB |
Output is correct |
65 |
Correct |
307 ms |
85240 KB |
Output is correct |
66 |
Correct |
675 ms |
169264 KB |
Output is correct |
67 |
Correct |
721 ms |
172448 KB |
Output is correct |
68 |
Correct |
1697 ms |
290512 KB |
Output is correct |
69 |
Correct |
2467 ms |
290516 KB |
Output is correct |
70 |
Correct |
1673 ms |
290416 KB |
Output is correct |
71 |
Correct |
1672 ms |
290612 KB |
Output is correct |
72 |
Execution timed out |
5085 ms |
825316 KB |
Time limit exceeded |
73 |
Halted |
0 ms |
0 KB |
- |