#include "bits/stdc++.h"
#include "rect.h"
using namespace std;
const int maxn = 2505;
struct rect {
int i, j;
int l, r;
rect (int i, int j, int l, int r) : i(i), j(j), l(l), r(r) {}
rect () {}
};
vector <pair <int, int>> ver[maxn][maxn], hor[maxn][maxn];
int t[maxn];
vector <int> v[maxn];
vector <rect> getLines(vector<vector<int>> &a) {
int n = a.size();
int m = a.front().size();
vector <rect> vert;
for(int i = 0; i < m; i++) v[i].clear();
set <int> last, aux;
vector <int> arr (n);
for(int i = 0; i < n; i++) {
last.clear();
fill(arr.begin(), arr.end(), -1);
for(int j = 0; j < m; j++) {
aux.clear();
for(int k = int(v[j].size()) - 1; k >= 0; k--) {
aux.insert(v[j][k]);
if(a[v[j][k]][j] >= a[i][j]) break;
}
while(!v[j].empty() && a[v[j].back()][j] <= a[i][j]) {
v[j].pop_back();
}
v[j].push_back(i);
for(int k : last) {
if(aux.find(k) == aux.end()) {
if(k < i - 1) vert.emplace_back(k, i, arr[k], j - 1);
arr[k] = -1;
}
}
for(int k : aux) {
if(arr[k] == -1) arr[k] = j;
}
last = aux;
}
for(int k : last) {
if(k < i - 1) vert.emplace_back(k, i, arr[k], m - 1);
}
}
return vert;
}
void rotate(vector <vector <int>> &a) {
int n = a.size();
int m = a.front().size();
vector <vector <int>> b (m, vector <int> (n, 0));
for(int i = 0; i < m; i++) {
for(int j = 0; j < n; j++) {
b[i][j] = a[j][i];
}
}
a = b;
}
void update(int x, int val) {
x += 1;
for(int i = x; i < maxn; i += i & (-i)) {
t[i] += val;
}
}
int query(int x) {
x += 1;
int sum = 0;
for(int i = x; i > 0; i -= i & (-i)) {
sum += t[i];
}
return sum;
}
long long solve(int i, int j) {
long long ans = 0;
auto cmp = [&] (pair <int, int> i, pair <int, int> j) {
return i.second < j.second;
};
sort(hor[i][j].begin(), hor[i][j].end(), cmp);
sort(ver[i][j].begin(), ver[i][j].end(), cmp);
int pt = 0;
for(auto y : ver[i][j]) {
while(pt < hor[i][j].size() && hor[i][j][pt].second <= y.second) {
update(hor[i][j][pt].first, 1);
++pt;
}
ans += query(2500) - query(y.first - 1);
}
for(int y = 0; y < pt; y++) update(hor[i][j][y].first, -1);
return ans;
}
long long count_rectangles(vector <vector <int>> a) {
int n = a.size();
int m = a.front().size();
auto h = getLines(a);
rotate(a);
auto v = getLines(a);
for(auto &x : v) {
int aux = x.l;
x.l = max(0, x.l - 1);
x.r = min(n - 1, x.r + 1);
for(int i = x.l; i <= x.r; i++) {
ver[i][x.j].emplace_back(x.l, x.i);
}
}
for(auto &x : h) {
int aux = x.l;
x.l = max(0, x.l - 1);
x.r = min(m - 1, x.r + 1);
for(int i = x.l; i <= x.r; i++) {
hor[x.j][i].emplace_back(x.i, x.l);
}
}
long long ans = 0;
for(int i = 0; i < n; i++) {
for(int j = 0; j < m; j++) {
ans += solve(i, j);
}
}
return ans;
}
Compilation message
rect.cpp: In function 'long long int solve(int, int)':
rect.cpp:86:14: warning: comparison of integer expressions of different signedness: 'int' and 'std::vector<std::pair<int, int> >::size_type' {aka 'long unsigned int'} [-Wsign-compare]
86 | while(pt < hor[i][j].size() && hor[i][j][pt].second <= y.second) {
| ~~~^~~~~~~~~~~~~~~~~~
rect.cpp: In function 'long long int count_rectangles(std::vector<std::vector<int> >)':
rect.cpp:102:9: warning: unused variable 'aux' [-Wunused-variable]
102 | int aux = x.l;
| ^~~
rect.cpp:110:9: warning: unused variable 'aux' [-Wunused-variable]
110 | int aux = x.l;
| ^~~
# |
Verdict |
Execution time |
Memory |
Grader output |
1 |
Correct |
186 ms |
295032 KB |
Output is correct |
2 |
Correct |
197 ms |
295160 KB |
Output is correct |
3 |
Correct |
189 ms |
295160 KB |
Output is correct |
4 |
Correct |
217 ms |
295160 KB |
Output is correct |
5 |
Correct |
217 ms |
295160 KB |
Output is correct |
6 |
Correct |
182 ms |
295160 KB |
Output is correct |
7 |
Correct |
211 ms |
295160 KB |
Output is correct |
8 |
Correct |
181 ms |
295160 KB |
Output is correct |
9 |
Correct |
180 ms |
295160 KB |
Output is correct |
10 |
Correct |
182 ms |
295160 KB |
Output is correct |
11 |
Correct |
185 ms |
295160 KB |
Output is correct |
12 |
Correct |
192 ms |
295144 KB |
Output is correct |
13 |
Correct |
207 ms |
295068 KB |
Output is correct |
14 |
Correct |
188 ms |
295032 KB |
Output is correct |
15 |
Correct |
195 ms |
295124 KB |
Output is correct |
16 |
Correct |
186 ms |
295032 KB |
Output is correct |
17 |
Correct |
188 ms |
295032 KB |
Output is correct |
18 |
Correct |
213 ms |
295028 KB |
Output is correct |
19 |
Correct |
194 ms |
295160 KB |
Output is correct |
20 |
Correct |
212 ms |
295032 KB |
Output is correct |
21 |
Correct |
181 ms |
295108 KB |
Output is correct |
# |
Verdict |
Execution time |
Memory |
Grader output |
1 |
Correct |
186 ms |
295032 KB |
Output is correct |
2 |
Correct |
197 ms |
295160 KB |
Output is correct |
3 |
Correct |
189 ms |
295160 KB |
Output is correct |
4 |
Correct |
217 ms |
295160 KB |
Output is correct |
5 |
Correct |
217 ms |
295160 KB |
Output is correct |
6 |
Correct |
182 ms |
295160 KB |
Output is correct |
7 |
Correct |
211 ms |
295160 KB |
Output is correct |
8 |
Correct |
181 ms |
295160 KB |
Output is correct |
9 |
Correct |
180 ms |
295160 KB |
Output is correct |
10 |
Correct |
182 ms |
295160 KB |
Output is correct |
11 |
Correct |
185 ms |
295160 KB |
Output is correct |
12 |
Correct |
192 ms |
295144 KB |
Output is correct |
13 |
Correct |
207 ms |
295068 KB |
Output is correct |
14 |
Correct |
188 ms |
295032 KB |
Output is correct |
15 |
Correct |
195 ms |
295124 KB |
Output is correct |
16 |
Correct |
186 ms |
295032 KB |
Output is correct |
17 |
Correct |
230 ms |
295416 KB |
Output is correct |
18 |
Correct |
205 ms |
295344 KB |
Output is correct |
19 |
Correct |
189 ms |
295360 KB |
Output is correct |
20 |
Correct |
182 ms |
295544 KB |
Output is correct |
21 |
Correct |
189 ms |
295800 KB |
Output is correct |
22 |
Correct |
188 ms |
295800 KB |
Output is correct |
23 |
Correct |
186 ms |
295800 KB |
Output is correct |
24 |
Correct |
201 ms |
295452 KB |
Output is correct |
25 |
Correct |
188 ms |
295032 KB |
Output is correct |
26 |
Correct |
213 ms |
295028 KB |
Output is correct |
27 |
Correct |
194 ms |
295160 KB |
Output is correct |
28 |
Correct |
212 ms |
295032 KB |
Output is correct |
29 |
Correct |
181 ms |
295108 KB |
Output is correct |
# |
Verdict |
Execution time |
Memory |
Grader output |
1 |
Correct |
186 ms |
295032 KB |
Output is correct |
2 |
Correct |
197 ms |
295160 KB |
Output is correct |
3 |
Correct |
189 ms |
295160 KB |
Output is correct |
4 |
Correct |
217 ms |
295160 KB |
Output is correct |
5 |
Correct |
217 ms |
295160 KB |
Output is correct |
6 |
Correct |
182 ms |
295160 KB |
Output is correct |
7 |
Correct |
211 ms |
295160 KB |
Output is correct |
8 |
Correct |
181 ms |
295160 KB |
Output is correct |
9 |
Correct |
180 ms |
295160 KB |
Output is correct |
10 |
Correct |
182 ms |
295160 KB |
Output is correct |
11 |
Correct |
185 ms |
295160 KB |
Output is correct |
12 |
Correct |
192 ms |
295144 KB |
Output is correct |
13 |
Correct |
207 ms |
295068 KB |
Output is correct |
14 |
Correct |
188 ms |
295032 KB |
Output is correct |
15 |
Correct |
195 ms |
295124 KB |
Output is correct |
16 |
Correct |
186 ms |
295032 KB |
Output is correct |
17 |
Correct |
230 ms |
295416 KB |
Output is correct |
18 |
Correct |
205 ms |
295344 KB |
Output is correct |
19 |
Correct |
189 ms |
295360 KB |
Output is correct |
20 |
Correct |
182 ms |
295544 KB |
Output is correct |
21 |
Correct |
189 ms |
295800 KB |
Output is correct |
22 |
Correct |
188 ms |
295800 KB |
Output is correct |
23 |
Correct |
186 ms |
295800 KB |
Output is correct |
24 |
Correct |
201 ms |
295452 KB |
Output is correct |
25 |
Correct |
232 ms |
296664 KB |
Output is correct |
26 |
Correct |
205 ms |
296660 KB |
Output is correct |
27 |
Correct |
229 ms |
296788 KB |
Output is correct |
28 |
Correct |
207 ms |
297808 KB |
Output is correct |
29 |
Correct |
287 ms |
299984 KB |
Output is correct |
30 |
Correct |
228 ms |
299912 KB |
Output is correct |
31 |
Correct |
217 ms |
299264 KB |
Output is correct |
32 |
Correct |
215 ms |
299220 KB |
Output is correct |
33 |
Correct |
188 ms |
295032 KB |
Output is correct |
34 |
Correct |
213 ms |
295028 KB |
Output is correct |
35 |
Correct |
194 ms |
295160 KB |
Output is correct |
36 |
Correct |
212 ms |
295032 KB |
Output is correct |
37 |
Correct |
181 ms |
295108 KB |
Output is correct |
# |
Verdict |
Execution time |
Memory |
Grader output |
1 |
Correct |
186 ms |
295032 KB |
Output is correct |
2 |
Correct |
197 ms |
295160 KB |
Output is correct |
3 |
Correct |
189 ms |
295160 KB |
Output is correct |
4 |
Correct |
217 ms |
295160 KB |
Output is correct |
5 |
Correct |
217 ms |
295160 KB |
Output is correct |
6 |
Correct |
182 ms |
295160 KB |
Output is correct |
7 |
Correct |
211 ms |
295160 KB |
Output is correct |
8 |
Correct |
181 ms |
295160 KB |
Output is correct |
9 |
Correct |
180 ms |
295160 KB |
Output is correct |
10 |
Correct |
182 ms |
295160 KB |
Output is correct |
11 |
Correct |
185 ms |
295160 KB |
Output is correct |
12 |
Correct |
192 ms |
295144 KB |
Output is correct |
13 |
Correct |
207 ms |
295068 KB |
Output is correct |
14 |
Correct |
188 ms |
295032 KB |
Output is correct |
15 |
Correct |
195 ms |
295124 KB |
Output is correct |
16 |
Correct |
186 ms |
295032 KB |
Output is correct |
17 |
Correct |
230 ms |
295416 KB |
Output is correct |
18 |
Correct |
205 ms |
295344 KB |
Output is correct |
19 |
Correct |
189 ms |
295360 KB |
Output is correct |
20 |
Correct |
182 ms |
295544 KB |
Output is correct |
21 |
Correct |
189 ms |
295800 KB |
Output is correct |
22 |
Correct |
188 ms |
295800 KB |
Output is correct |
23 |
Correct |
186 ms |
295800 KB |
Output is correct |
24 |
Correct |
201 ms |
295452 KB |
Output is correct |
25 |
Correct |
232 ms |
296664 KB |
Output is correct |
26 |
Correct |
205 ms |
296660 KB |
Output is correct |
27 |
Correct |
229 ms |
296788 KB |
Output is correct |
28 |
Correct |
207 ms |
297808 KB |
Output is correct |
29 |
Correct |
287 ms |
299984 KB |
Output is correct |
30 |
Correct |
228 ms |
299912 KB |
Output is correct |
31 |
Correct |
217 ms |
299264 KB |
Output is correct |
32 |
Correct |
215 ms |
299220 KB |
Output is correct |
33 |
Correct |
432 ms |
325888 KB |
Output is correct |
34 |
Correct |
431 ms |
328800 KB |
Output is correct |
35 |
Correct |
433 ms |
328704 KB |
Output is correct |
36 |
Correct |
464 ms |
331616 KB |
Output is correct |
37 |
Correct |
496 ms |
313748 KB |
Output is correct |
38 |
Correct |
475 ms |
313756 KB |
Output is correct |
39 |
Correct |
495 ms |
313752 KB |
Output is correct |
40 |
Correct |
515 ms |
313208 KB |
Output is correct |
41 |
Correct |
390 ms |
320872 KB |
Output is correct |
42 |
Correct |
476 ms |
327516 KB |
Output is correct |
43 |
Correct |
670 ms |
351472 KB |
Output is correct |
44 |
Correct |
785 ms |
351968 KB |
Output is correct |
45 |
Correct |
424 ms |
323824 KB |
Output is correct |
46 |
Correct |
427 ms |
323772 KB |
Output is correct |
47 |
Correct |
657 ms |
342232 KB |
Output is correct |
48 |
Correct |
611 ms |
342808 KB |
Output is correct |
49 |
Correct |
188 ms |
295032 KB |
Output is correct |
50 |
Correct |
213 ms |
295028 KB |
Output is correct |
51 |
Correct |
194 ms |
295160 KB |
Output is correct |
52 |
Correct |
212 ms |
295032 KB |
Output is correct |
53 |
Correct |
181 ms |
295108 KB |
Output is correct |
# |
Verdict |
Execution time |
Memory |
Grader output |
1 |
Correct |
204 ms |
295836 KB |
Output is correct |
2 |
Correct |
186 ms |
295672 KB |
Output is correct |
3 |
Correct |
189 ms |
295544 KB |
Output is correct |
4 |
Correct |
185 ms |
295160 KB |
Output is correct |
5 |
Correct |
202 ms |
295928 KB |
Output is correct |
6 |
Correct |
222 ms |
295800 KB |
Output is correct |
7 |
Correct |
222 ms |
295796 KB |
Output is correct |
8 |
Correct |
195 ms |
295800 KB |
Output is correct |
9 |
Correct |
221 ms |
295800 KB |
Output is correct |
10 |
Correct |
218 ms |
295416 KB |
Output is correct |
11 |
Correct |
214 ms |
295544 KB |
Output is correct |
# |
Verdict |
Execution time |
Memory |
Grader output |
1 |
Correct |
184 ms |
295032 KB |
Output is correct |
2 |
Correct |
1514 ms |
442956 KB |
Output is correct |
3 |
Correct |
3357 ms |
615972 KB |
Output is correct |
4 |
Correct |
3300 ms |
617120 KB |
Output is correct |
5 |
Correct |
3288 ms |
617288 KB |
Output is correct |
6 |
Correct |
755 ms |
344184 KB |
Output is correct |
7 |
Correct |
1292 ms |
366716 KB |
Output is correct |
8 |
Correct |
1364 ms |
369784 KB |
Output is correct |
9 |
Correct |
188 ms |
295032 KB |
Output is correct |
10 |
Correct |
213 ms |
295028 KB |
Output is correct |
11 |
Correct |
194 ms |
295160 KB |
Output is correct |
12 |
Correct |
212 ms |
295032 KB |
Output is correct |
13 |
Correct |
181 ms |
295108 KB |
Output is correct |
# |
Verdict |
Execution time |
Memory |
Grader output |
1 |
Correct |
186 ms |
295032 KB |
Output is correct |
2 |
Correct |
197 ms |
295160 KB |
Output is correct |
3 |
Correct |
189 ms |
295160 KB |
Output is correct |
4 |
Correct |
217 ms |
295160 KB |
Output is correct |
5 |
Correct |
217 ms |
295160 KB |
Output is correct |
6 |
Correct |
182 ms |
295160 KB |
Output is correct |
7 |
Correct |
211 ms |
295160 KB |
Output is correct |
8 |
Correct |
181 ms |
295160 KB |
Output is correct |
9 |
Correct |
180 ms |
295160 KB |
Output is correct |
10 |
Correct |
182 ms |
295160 KB |
Output is correct |
11 |
Correct |
185 ms |
295160 KB |
Output is correct |
12 |
Correct |
192 ms |
295144 KB |
Output is correct |
13 |
Correct |
207 ms |
295068 KB |
Output is correct |
14 |
Correct |
188 ms |
295032 KB |
Output is correct |
15 |
Correct |
195 ms |
295124 KB |
Output is correct |
16 |
Correct |
186 ms |
295032 KB |
Output is correct |
17 |
Correct |
230 ms |
295416 KB |
Output is correct |
18 |
Correct |
205 ms |
295344 KB |
Output is correct |
19 |
Correct |
189 ms |
295360 KB |
Output is correct |
20 |
Correct |
182 ms |
295544 KB |
Output is correct |
21 |
Correct |
189 ms |
295800 KB |
Output is correct |
22 |
Correct |
188 ms |
295800 KB |
Output is correct |
23 |
Correct |
186 ms |
295800 KB |
Output is correct |
24 |
Correct |
201 ms |
295452 KB |
Output is correct |
25 |
Correct |
232 ms |
296664 KB |
Output is correct |
26 |
Correct |
205 ms |
296660 KB |
Output is correct |
27 |
Correct |
229 ms |
296788 KB |
Output is correct |
28 |
Correct |
207 ms |
297808 KB |
Output is correct |
29 |
Correct |
287 ms |
299984 KB |
Output is correct |
30 |
Correct |
228 ms |
299912 KB |
Output is correct |
31 |
Correct |
217 ms |
299264 KB |
Output is correct |
32 |
Correct |
215 ms |
299220 KB |
Output is correct |
33 |
Correct |
432 ms |
325888 KB |
Output is correct |
34 |
Correct |
431 ms |
328800 KB |
Output is correct |
35 |
Correct |
433 ms |
328704 KB |
Output is correct |
36 |
Correct |
464 ms |
331616 KB |
Output is correct |
37 |
Correct |
496 ms |
313748 KB |
Output is correct |
38 |
Correct |
475 ms |
313756 KB |
Output is correct |
39 |
Correct |
495 ms |
313752 KB |
Output is correct |
40 |
Correct |
515 ms |
313208 KB |
Output is correct |
41 |
Correct |
390 ms |
320872 KB |
Output is correct |
42 |
Correct |
476 ms |
327516 KB |
Output is correct |
43 |
Correct |
670 ms |
351472 KB |
Output is correct |
44 |
Correct |
785 ms |
351968 KB |
Output is correct |
45 |
Correct |
424 ms |
323824 KB |
Output is correct |
46 |
Correct |
427 ms |
323772 KB |
Output is correct |
47 |
Correct |
657 ms |
342232 KB |
Output is correct |
48 |
Correct |
611 ms |
342808 KB |
Output is correct |
49 |
Correct |
204 ms |
295836 KB |
Output is correct |
50 |
Correct |
186 ms |
295672 KB |
Output is correct |
51 |
Correct |
189 ms |
295544 KB |
Output is correct |
52 |
Correct |
185 ms |
295160 KB |
Output is correct |
53 |
Correct |
202 ms |
295928 KB |
Output is correct |
54 |
Correct |
222 ms |
295800 KB |
Output is correct |
55 |
Correct |
222 ms |
295796 KB |
Output is correct |
56 |
Correct |
195 ms |
295800 KB |
Output is correct |
57 |
Correct |
221 ms |
295800 KB |
Output is correct |
58 |
Correct |
218 ms |
295416 KB |
Output is correct |
59 |
Correct |
214 ms |
295544 KB |
Output is correct |
60 |
Correct |
184 ms |
295032 KB |
Output is correct |
61 |
Correct |
1514 ms |
442956 KB |
Output is correct |
62 |
Correct |
3357 ms |
615972 KB |
Output is correct |
63 |
Correct |
3300 ms |
617120 KB |
Output is correct |
64 |
Correct |
3288 ms |
617288 KB |
Output is correct |
65 |
Correct |
755 ms |
344184 KB |
Output is correct |
66 |
Correct |
1292 ms |
366716 KB |
Output is correct |
67 |
Correct |
1364 ms |
369784 KB |
Output is correct |
68 |
Correct |
3934 ms |
655636 KB |
Output is correct |
69 |
Correct |
4335 ms |
704244 KB |
Output is correct |
70 |
Correct |
3785 ms |
706452 KB |
Output is correct |
71 |
Correct |
4148 ms |
753268 KB |
Output is correct |
72 |
Execution timed out |
5079 ms |
521448 KB |
Time limit exceeded |
73 |
Halted |
0 ms |
0 KB |
- |