#include "rainbow.h"
#include <bits/stdc++.h>
using namespace std;
#define debug(x) cerr << #x << " = " << x << '\n';
#define BP() cerr << "OK!\n";
#define PR(A, n) {cerr << #A << " = "; for (int _=1; _<=n; ++_) cerr << A[_] << ' '; cerr << '\n';}
#define PR0(A, n) {cerr << #A << " = "; for (int _=0; _<n; ++_) cerr << A[_] << ' '; cerr << '\n';}
#define FILE_NAME "data"
const int X[] = {1, 0, -1, 0};
const int Y[] = {0, 1, 0, -1};
const int INF = 1e9;
struct fenwick_tree_2D {
int m, n;
vector<vector<int> > bit;
void init(int _m, int _n) {
m = _m;
n = _n;
bit.resize(m+1);
for (int i=0; i<=m; ++i)
bit[i].resize(n+1);
}
void add(int x, int y) {
if (!bit[x][y])
++bit[x][y];
}
void process() {
for (int i=1; i<=m; ++i) {
for (int j=1; j<=n; ++j)
bit[i][j] += bit[i-1][j] + bit[i][j-1] - bit[i-1][j-1];
}
}
int get(int x1, int y1, int x2, int y2) {
if (x1>x2 || y1>y2)
return 0;
return bit[x2][y2] - bit[x2][y1-1] - bit[x1-1][y2] + bit[x1-1][y1-1];
}
};
int m, n, min_x, max_x, min_y, max_y;
vector<pair<int, int> > snake;
bool visited[52][52];
fenwick_tree_2D T1, T2, R, C;
void add(pair<int, int> p) {
T1.add(p.first, p.second);
for (int i=0; i<4; ++i) {
T2.add(p.first, p.second);
p.first += X[i];
p.second += Y[i];
}
C.add(p.first, p.second);
C.add(p.first, p.second+1);
R.add(p.first, p.second);
R.add(p.first+1, p.second);
}
void init(int _R, int _C, int sr, int sc, int M, char *S) {
m = _R;
n = _C;
max_x = max_y = -INF;
min_x = min_y = INF;
snake.resize(M+1);
snake[0] = {sr, sc};
for (int i=1; i<=M; ++i) {
snake[i] = snake[i-1];
if (S[i-1]=='N')
--snake[i].first;
else if (S[i-1]=='S')
++snake[i].first;
else if (S[i-1]=='W')
--snake[i].second;
else
++snake[i].second;
}
sort(snake.begin(), snake.end());
snake.resize(unique(snake.begin(), snake.end()) - snake.begin());
T1.init(m+1, n+1);
T2.init(m+1, n+1);
R.init(m+1, n+1);
C.init(m+1, n+1);
for (auto p : snake) {
min_x = min(min_x, p.first);
max_x = max(max_x, p.first);
min_y = min(min_y, p.second);
max_y = max(max_y, p.second);
add(p);
}
T1.process();
T2.process();
R.process();
C.process();
}
int colour(int x1, int y1, int x2, int y2) {
int v = T2.get(x1+1, y1+1, x2, y2);
int e = R.get(x1+1, y1, x2, y2) + C.get(x1, y1+1, x2, y2);
// debug(R.get(x1+1, y1+1, x2, y2-1));
int res = e - v + 1;
if (x1<min_x && max_x<x2 && y1<min_y && max_y<y2)
++res;
res -= T1.get(x1, y1, x2, y2);
return res;
}
# |
결과 |
실행 시간 |
메모리 |
Grader output |
1 |
Correct |
3 ms |
384 KB |
Output is correct |
2 |
Correct |
3 ms |
384 KB |
Output is correct |
3 |
Correct |
4 ms |
384 KB |
Output is correct |
4 |
Correct |
4 ms |
384 KB |
Output is correct |
5 |
Correct |
4 ms |
384 KB |
Output is correct |
6 |
Correct |
4 ms |
256 KB |
Output is correct |
7 |
Correct |
3 ms |
384 KB |
Output is correct |
8 |
Correct |
4 ms |
384 KB |
Output is correct |
9 |
Correct |
2 ms |
384 KB |
Output is correct |
10 |
Correct |
3 ms |
384 KB |
Output is correct |
11 |
Correct |
4 ms |
384 KB |
Output is correct |
12 |
Correct |
5 ms |
384 KB |
Output is correct |
13 |
Correct |
4 ms |
384 KB |
Output is correct |
14 |
Correct |
4 ms |
512 KB |
Output is correct |
15 |
Correct |
2 ms |
256 KB |
Output is correct |
16 |
Correct |
3 ms |
384 KB |
Output is correct |
17 |
Correct |
2 ms |
256 KB |
Output is correct |
# |
결과 |
실행 시간 |
메모리 |
Grader output |
1 |
Correct |
3 ms |
384 KB |
Output is correct |
2 |
Correct |
2 ms |
256 KB |
Output is correct |
3 |
Correct |
154 ms |
16860 KB |
Output is correct |
4 |
Correct |
161 ms |
17372 KB |
Output is correct |
5 |
Correct |
149 ms |
17272 KB |
Output is correct |
6 |
Correct |
123 ms |
17272 KB |
Output is correct |
7 |
Correct |
142 ms |
17424 KB |
Output is correct |
8 |
Correct |
154 ms |
17144 KB |
Output is correct |
9 |
Correct |
149 ms |
17272 KB |
Output is correct |
10 |
Correct |
172 ms |
17400 KB |
Output is correct |
11 |
Correct |
155 ms |
17344 KB |
Output is correct |
12 |
Correct |
104 ms |
17144 KB |
Output is correct |
13 |
Correct |
104 ms |
17272 KB |
Output is correct |
14 |
Correct |
113 ms |
17276 KB |
Output is correct |
15 |
Correct |
94 ms |
17400 KB |
Output is correct |
16 |
Correct |
117 ms |
17116 KB |
Output is correct |
# |
결과 |
실행 시간 |
메모리 |
Grader output |
1 |
Correct |
2 ms |
256 KB |
Output is correct |
2 |
Runtime error |
1552 ms |
1049600 KB |
Execution killed with signal 9 (could be triggered by violating memory limits) |
3 |
Halted |
0 ms |
0 KB |
- |
# |
결과 |
실행 시간 |
메모리 |
Grader output |
1 |
Correct |
3 ms |
384 KB |
Output is correct |
2 |
Correct |
3 ms |
384 KB |
Output is correct |
3 |
Correct |
4 ms |
384 KB |
Output is correct |
4 |
Correct |
4 ms |
384 KB |
Output is correct |
5 |
Correct |
4 ms |
384 KB |
Output is correct |
6 |
Correct |
4 ms |
256 KB |
Output is correct |
7 |
Correct |
3 ms |
384 KB |
Output is correct |
8 |
Correct |
4 ms |
384 KB |
Output is correct |
9 |
Correct |
2 ms |
384 KB |
Output is correct |
10 |
Correct |
3 ms |
384 KB |
Output is correct |
11 |
Correct |
4 ms |
384 KB |
Output is correct |
12 |
Correct |
5 ms |
384 KB |
Output is correct |
13 |
Correct |
4 ms |
384 KB |
Output is correct |
14 |
Correct |
4 ms |
512 KB |
Output is correct |
15 |
Correct |
2 ms |
256 KB |
Output is correct |
16 |
Correct |
3 ms |
384 KB |
Output is correct |
17 |
Correct |
2 ms |
256 KB |
Output is correct |
18 |
Correct |
175 ms |
20540 KB |
Output is correct |
19 |
Correct |
90 ms |
5496 KB |
Output is correct |
20 |
Correct |
69 ms |
3976 KB |
Output is correct |
21 |
Correct |
67 ms |
4472 KB |
Output is correct |
22 |
Correct |
81 ms |
5496 KB |
Output is correct |
23 |
Correct |
131 ms |
5404 KB |
Output is correct |
24 |
Correct |
65 ms |
4088 KB |
Output is correct |
25 |
Correct |
99 ms |
4476 KB |
Output is correct |
26 |
Correct |
90 ms |
5380 KB |
Output is correct |
27 |
Correct |
141 ms |
20600 KB |
Output is correct |
28 |
Correct |
160 ms |
20580 KB |
Output is correct |
29 |
Correct |
164 ms |
20540 KB |
Output is correct |
30 |
Correct |
162 ms |
20600 KB |
Output is correct |
31 |
Correct |
6 ms |
512 KB |
Output is correct |
32 |
Correct |
113 ms |
20600 KB |
Output is correct |
# |
결과 |
실행 시간 |
메모리 |
Grader output |
1 |
Correct |
3 ms |
384 KB |
Output is correct |
2 |
Correct |
3 ms |
384 KB |
Output is correct |
3 |
Correct |
4 ms |
384 KB |
Output is correct |
4 |
Correct |
4 ms |
384 KB |
Output is correct |
5 |
Correct |
4 ms |
384 KB |
Output is correct |
6 |
Correct |
4 ms |
256 KB |
Output is correct |
7 |
Correct |
3 ms |
384 KB |
Output is correct |
8 |
Correct |
4 ms |
384 KB |
Output is correct |
9 |
Correct |
2 ms |
384 KB |
Output is correct |
10 |
Correct |
3 ms |
384 KB |
Output is correct |
11 |
Correct |
4 ms |
384 KB |
Output is correct |
12 |
Correct |
5 ms |
384 KB |
Output is correct |
13 |
Correct |
4 ms |
384 KB |
Output is correct |
14 |
Correct |
4 ms |
512 KB |
Output is correct |
15 |
Correct |
2 ms |
256 KB |
Output is correct |
16 |
Correct |
3 ms |
384 KB |
Output is correct |
17 |
Correct |
2 ms |
256 KB |
Output is correct |
18 |
Correct |
175 ms |
20540 KB |
Output is correct |
19 |
Correct |
90 ms |
5496 KB |
Output is correct |
20 |
Correct |
69 ms |
3976 KB |
Output is correct |
21 |
Correct |
67 ms |
4472 KB |
Output is correct |
22 |
Correct |
81 ms |
5496 KB |
Output is correct |
23 |
Correct |
131 ms |
5404 KB |
Output is correct |
24 |
Correct |
65 ms |
4088 KB |
Output is correct |
25 |
Correct |
99 ms |
4476 KB |
Output is correct |
26 |
Correct |
90 ms |
5380 KB |
Output is correct |
27 |
Correct |
141 ms |
20600 KB |
Output is correct |
28 |
Correct |
160 ms |
20580 KB |
Output is correct |
29 |
Correct |
164 ms |
20540 KB |
Output is correct |
30 |
Correct |
162 ms |
20600 KB |
Output is correct |
31 |
Correct |
6 ms |
512 KB |
Output is correct |
32 |
Correct |
113 ms |
20600 KB |
Output is correct |
33 |
Runtime error |
1552 ms |
1049600 KB |
Execution killed with signal 9 (could be triggered by violating memory limits) |
34 |
Halted |
0 ms |
0 KB |
- |