#include "rainbow.h"
#include <iostream>
#include <vector>
#include <cassert>
#include <algorithm>
#include <set>
struct Brute {
std::set<std::pair<int, int>> st;
void add(int x, int y) {
st.insert({x, y});
}
int query(int x1, int y1, int x2, int y2) {
int ret = 0;
for (const auto &[x, y] : st) {
if (x1 <= x && x <= x2 && y1 <= y && y <= y2) {
ret++;
}
}
return ret;
}
};
namespace {
std::set<std::pair<int, int>> cells;
int minX, maxX, minY, maxY;
Brute vert, horizontalEdges, verticalEdges, boxes;
};
void init(int R, int C, int sr, int sc, int M, char *S) {
int x = sr, y = sc;
minX = R + 1;
minY = C + 1;
maxX = 0;
maxY = 0;
auto add = [&](int x, int y) {
cells.insert({x, y});
minX = std::min(minX, x);
maxX = std::max(maxX, x);
minY = std::min(minY, y);
maxY = std::max(maxY, y);
vert.add(x, y);
};
add(x, y);
for (int i = 0; i < M; i++) {
if (S[i] == 'N') {
x--;
} else if (S[i] == 'S') {
x++;
} else if (S[i] == 'W') {
y--;
} else {
y++;
}
add(x, y);
horizontalEdges.add(x, y);
horizontalEdges.add(x, y - 1);
verticalEdges.add(x, y);
verticalEdges.add(x - 1, y);
boxes.add(x, y);
boxes.add(x, y - 1);
boxes.add(x - 1, y);
boxes.add(x - 1, y - 1);
}
}
int colour(int x1, int y1, int x2, int y2) {
int V = (x2 - x1 + 1) * (y2 - y1 + 1) - vert.query(x1, y1, x2, y2);
int E = 0;
E += (x2 - x1 + 1) * (y2 - y1) - horizontalEdges.query(x1, y1, x2, y2 - 1);
E += (x2 - x1) * (y2 - y1 + 1) - verticalEdges.query(x1, y1, x2 - 1, y2);
int F = (x2 - x1) * (y2 - y1) - boxes.query(x1, y1, x2 - 1, y2 - 1);
F++;
if (x1 < minX && maxX < x2 && y1 < minY && maxY < y2) {
F++;
}
// V + F = E + C + 1 (cred)
return V + F - E - 1;
}
