#include "rainbow.h"
#include <bits/stdc++.h>
using namespace std;
const int N = 1002;
const int maxN = (int) 2e5 + 9;
int sub = 1;
int a[N][N], vis[N][N], done[N][N], u, l, d, r, sum1[maxN], sum2[maxN], sum12[maxN], vis2[2][maxN];
array<int, 4> di = {0, 0, 1, -1};
array<int, 4> dj = {1, -1, 0, 0};
bool valid(int i, int j) {
return (u <= i && i <= d && l <= j && j <= r && !vis[i][j] && !done[i][j]);
}
void dfs(int i, int j) {
done[i][j] = 1;
for (int k = 0; k < 4; k++) {
int ni = i + di[k];
int nj = j + dj[k];
if (valid(ni, nj)) {
dfs(ni, nj);
}
}
}
int isNew(int i) {
if ((!vis2[0][i] || !vis2[1][i]) && (vis2[0][i - 1] && vis2[1][i - 1])) {
return 1;
}
if (vis2[0][i] != vis2[0][i - 1] && vis2[1][i] != vis2[1][i - 1]) {
return 1;
}
return 0;
}
void init(int R, int C, int sr, int sc, int M, char *S) {
if (R == 2) {
sub = 2;
}
u = 1; d = R;
l = 1; r = C;
if (sub == 2) {
vis2[sr][sc] = 1;
} else {
vis[sr][sc] = 1;
}
for (int i = 0; i < M; i++) {
if (S[i] == 'N') {
sr--;
} else if (S[i] == 'S') {
sr++;
} else if (S[i] == 'E') {
sc++;
} else {
sc--;
}
if (sub == 2) {
vis2[sr][sc] = 1;
} else {
vis[sr][sc] = 1;
}
}
if (sub == 2) {
vis2[0][0] = vis2[1][0] = 1;
for (int i = 1; i <= C; i++) {
sum1[i] = sum1[i - 1] + vis2[0][i];
sum2[i] = sum2[i - 1] + vis2[1][i];
sum12[i] = sum12[i - 1] + isNew(i);
}
}
}
int colour(int ar, int ac, int br, int bc) {
u = ar; d = br;
l = ac; r = bc;
int ans = 0;
if (sub == 2) {
int len = (r - l + 1);
if (u == d) {
if (d == 1) {
return len - (sum1[r] - sum1[l - 1]);
} else {
return len - (sum2[r] - sum2[l - 1]);
}
} else {
return (sum12[r] - sum12[l - 1]) + ((!vis2[0][l] || !vis2[1][l]) && !isNew(i));
}
}
for (int i = u; i <= d; i++) {
for (int j = l; j <= r; j++) {
if (!vis[i][j] && !done[i][j]) {
dfs(i, j);
ans++;
}
}
}
memset(done, 0, sizeof done);
return ans;
}
Compilation message
rainbow.cpp: In function 'int colour(int, int, int, int)':
rainbow.cpp:92:79: error: 'i' was not declared in this scope; did you mean 'di'?
92 | return (sum12[r] - sum12[l - 1]) + ((!vis2[0][l] || !vis2[1][l]) && !isNew(i));
| ^
| di
