#ifndef __RAINBOW_H
#define __RAINBOW_H
#include "rainbow.h"
#include <iostream>
#include <cassert>
#include <string>
#include <set>
#include <string>
#include <algorithm>
#define mp make_pair
#define x first
#define y second
using namespace std;
typedef pair<int, int> pii;
string dirs = "NSWE";
int id(char c) {
return find(dirs.begin(), dirs.end(), c) - dirs.begin();
}
pii shs[4] = {
mp(-1, 0),
mp(1, 0),
mp(0, -1),
mp(0, 1)
};
set<pii> edges[4];
set<pii> Qu;
set<pii> forb;
void init(int R, int C, int sr, int sc, int M, char *S) {
assert(forb.empty());
sr++, sc++;
forb.emplace(sr, sc);
for (int i = 0; i < M; i++) {
sr += shs[id(S[i])].x;
sc += shs[id(S[i])].y;
forb.emplace(sr, sc);
}
for (pii ceil : forb) {
if (forb.count(mp(ceil.x, ceil.y + 1)))
edges[0].insert(ceil);
if (forb.count(mp(ceil.x - 1, ceil.y)))
edges[2].insert(ceil);
if (forb.count(mp(ceil.x - 1, ceil.y + 1))
&& (!forb.count(mp(ceil.x - 1, ceil.y)) && !forb.count(mp(ceil.x, ceil.y + 1))))
edges[1].insert(ceil);
if (forb.count(mp(ceil.x - 1, ceil.y - 1))
&& (!forb.count(mp(ceil.x - 1, ceil.y)) && !forb.count(mp(ceil.x, ceil.y - 1))))
edges[3].insert(ceil);
bool fail = false;
for (pii sh : {
mp(1, 0),
mp(1, 1),
mp(0, 1)
}) {
if (!forb.count(mp(ceil.x + sh.x, ceil.y + sh.y)))
fail = true;
}
if (!fail) Qu.insert(ceil);
}
}
int colour(int ar, int ac, int br, int bc) {
ar++, ac++, br++, bc++;
int V = 0;
int E = 0;
auto in = [ar, ac, br, bc](pii ceil) {
return (ar <= ceil.x && ceil.x <= br
&& ac <= ceil.y && ceil.y <= bc);
};
for (pii fb : forb)
if (in(fb))
V++;
for (pii p : edges[0])
if (in(p) && in(mp(p.x, p.y + 1)))
E++;
for (pii p : edges[2])
if (in(p) && in(mp(p.x - 1, p.y)))
E++;
for (pii p : edges[1])
if (in(p) && in(mp(p.x - 1, p.y + 1)))
E++;
for (pii p : edges[3])
if (in(p) && in(mp(p.x - 1, p.y - 1)))
E++;
int Q = 0;
for (pii q : Qu)
if (in(q) && q.x < br && q.y < bc)
Q++;
int hit = 0;
int nonempty = 0;
for (pii fb : forb) {
if (in(fb)) {
nonempty = 1;
bool is_row = (fb.x == ar || fb.x == br);
bool is_col = (fb.y == ac || fb.y == bc);
if (is_row || is_col) hit = 1;
if (fb.x == ar) E++;
if (fb.x == br) E++;
if (fb.y == ac) E++;
if (fb.y == bc) E++;
if (is_row)
if (fb.y + 1 <= bc && forb.count(mp(fb.x, fb.y + 1)))
Q++;
if (is_col)
if (fb.x + 1 <= br && forb.count(mp(fb.x + 1, fb.y)))
Q++;
if (fb.x == ar && fb.y == ac) Q++;
if (fb.x == ar && fb.y == bc) Q++;
if (fb.x == br && fb.y == ac) Q++;
if (fb.x == br && fb.y == bc) Q++;
}
}
return 1 - V + E - Q + (1 - hit) - (1 - nonempty);
}
#endif // __RAINBOW_H
# |
Verdict |
Execution time |
Memory |
Grader output |
1 |
Incorrect |
7 ms |
376 KB |
Output isn't correct |
2 |
Halted |
0 ms |
0 KB |
- |
# |
Verdict |
Execution time |
Memory |
Grader output |
1 |
Incorrect |
2 ms |
256 KB |
Output isn't correct |
2 |
Halted |
0 ms |
0 KB |
- |
# |
Verdict |
Execution time |
Memory |
Grader output |
1 |
Correct |
2 ms |
256 KB |
Output is correct |
2 |
Correct |
157 ms |
9732 KB |
Output is correct |
3 |
Correct |
155 ms |
9740 KB |
Output is correct |
4 |
Correct |
172 ms |
9824 KB |
Output is correct |
5 |
Correct |
108 ms |
7480 KB |
Output is correct |
6 |
Correct |
49 ms |
4472 KB |
Output is correct |
7 |
Correct |
93 ms |
8568 KB |
Output is correct |
8 |
Correct |
94 ms |
8964 KB |
Output is correct |
9 |
Correct |
82 ms |
8136 KB |
Output is correct |
10 |
Correct |
42 ms |
3960 KB |
Output is correct |
11 |
Correct |
72 ms |
5112 KB |
Output is correct |
12 |
Correct |
149 ms |
9800 KB |
Output is correct |
13 |
Correct |
150 ms |
9848 KB |
Output is correct |
14 |
Correct |
163 ms |
9848 KB |
Output is correct |
15 |
Correct |
106 ms |
7416 KB |
Output is correct |
16 |
Correct |
42 ms |
3832 KB |
Output is correct |
17 |
Correct |
87 ms |
8440 KB |
Output is correct |
18 |
Correct |
98 ms |
9848 KB |
Output is correct |
19 |
Correct |
146 ms |
9840 KB |
Output is correct |
20 |
Correct |
157 ms |
9720 KB |
Output is correct |
21 |
Correct |
92 ms |
9080 KB |
Output is correct |
22 |
Correct |
88 ms |
8156 KB |
Output is correct |
23 |
Correct |
43 ms |
3832 KB |
Output is correct |
24 |
Correct |
66 ms |
5084 KB |
Output is correct |
25 |
Correct |
156 ms |
9812 KB |
Output is correct |
26 |
Correct |
151 ms |
9720 KB |
Output is correct |
27 |
Correct |
164 ms |
9848 KB |
Output is correct |
28 |
Correct |
109 ms |
7480 KB |
Output is correct |
29 |
Correct |
42 ms |
3832 KB |
Output is correct |
30 |
Correct |
88 ms |
8568 KB |
Output is correct |
31 |
Correct |
105 ms |
9780 KB |
Output is correct |
32 |
Correct |
157 ms |
9808 KB |
Output is correct |
33 |
Correct |
154 ms |
9748 KB |
Output is correct |
# |
Verdict |
Execution time |
Memory |
Grader output |
1 |
Incorrect |
7 ms |
376 KB |
Output isn't correct |
2 |
Halted |
0 ms |
0 KB |
- |
# |
Verdict |
Execution time |
Memory |
Grader output |
1 |
Incorrect |
7 ms |
376 KB |
Output isn't correct |
2 |
Halted |
0 ms |
0 KB |
- |