#include <cstdio>
#include <tuple>
#include <deque>
using namespace std;
const int MAXN = 404;
typedef tuple<bool,int,int> ii;
int N, M;
bool D[MAXN][MAXN];
int ans[MAXN][MAXN];
int sz[MAXN * MAXN];
ii visited[MAXN][MAXN];
char imsi[MAXN];
int dx[4] = {-1, 0, 1, 0}, dy[4] = {0, 1, 0, -1};
tuple<int,int> clear_div(int x, int y) {
if (x < 0) {
const int d = (-x) / y;
const int r = (-x) % y;
if (r == 0) return make_tuple(-d, 0);
else return make_tuple(-(d + 1), y - r);
}
return make_tuple(x / y, x % y);
}
int main() {
// 1. input
scanf("%d %d", &N, &M);
for (int i = 0; i < N; ++i) {
scanf("%s", imsi);
for (int j = 0; j < M; ++j) {
D[i][j] = (imsi[j] == 'B');
}
}
// 2. process
deque<tuple<int, int>> Q;
int cnt = 1;
for (int i = 0; i < N; ++i) {
for (int j = 0; j < M; ++j) {
if (ans[i][j]) continue;
bool color = D[i][j];
bool inf = false;
Q.clear();
Q.emplace_back(i, j);
visited[i][j] = make_tuple(true, 0, 0);
while(!Q.empty()) {
const int p = get<0>(Q.front());
const int q = get<1>(Q.front());
Q.pop_front();
++sz[cnt];
int pr, qr, pd, qd;
tie(pd, pr) = clear_div(p, N);
tie(qd, qr) = clear_div(q, M);
ans[pr][qr] = cnt;
for (int k = 0; k < 4; ++k) {
int xpr, xqr, xpd, xqd;
tie(xpd, xpr) = clear_div(p + dx[k], N);
tie(xqd, xqr) = clear_div(q + dy[k], M);
if (D[xpr][xqr] != color) continue;
else if (get<0>(visited[xpr][xqr])) {
if (xpd == get<1>(visited[xpr][xqr])\
&& xqd == get<2>(visited[xpr][xqr])) {
continue;
} else {
// problem!
inf = true;
}
} else {
visited[xpr][xqr] = make_tuple(true, xpd, xqd);
Q.emplace_back(p + dx[k], q + dy[k]);
}
}
}
if (inf) sz[cnt] = -1;
++cnt;
}
}
for (int i = 0; i < N; ++i) {
for (int j = 0; j < M; ++j) {
printf("%d ", sz[ans[i][j]]);
}
puts("");
}
return 0;
}
Compilation message
YZ.cpp: In function 'int main()':
YZ.cpp:31:24: warning: ignoring return value of 'int scanf(const char*, ...)', declared with attribute warn_unused_result [-Wunused-result]
scanf("%d %d", &N, &M);
^
YZ.cpp:33:20: warning: ignoring return value of 'int scanf(const char*, ...)', declared with attribute warn_unused_result [-Wunused-result]
scanf("%s", imsi);
^
# |
결과 |
실행 시간 |
메모리 |
Grader output |
1 |
Correct |
33 ms |
5284 KB |
Output is correct |
2 |
Correct |
29 ms |
5284 KB |
Output is correct |
3 |
Correct |
26 ms |
5284 KB |
Output is correct |
4 |
Correct |
29 ms |
5284 KB |
Output is correct |
5 |
Correct |
23 ms |
5284 KB |
Output is correct |
6 |
Correct |
29 ms |
5284 KB |
Output is correct |
7 |
Correct |
26 ms |
5284 KB |
Output is correct |
8 |
Correct |
29 ms |
5284 KB |
Output is correct |
9 |
Correct |
33 ms |
5284 KB |
Output is correct |
10 |
Correct |
26 ms |
5284 KB |
Output is correct |
11 |
Correct |
26 ms |
5284 KB |
Output is correct |
12 |
Correct |
19 ms |
5284 KB |
Output is correct |
13 |
Correct |
26 ms |
5284 KB |
Output is correct |
14 |
Correct |
19 ms |
5284 KB |
Output is correct |
15 |
Correct |
33 ms |
5284 KB |
Output is correct |
16 |
Correct |
36 ms |
5284 KB |
Output is correct |
17 |
Correct |
26 ms |
5284 KB |
Output is correct |
18 |
Correct |
33 ms |
5284 KB |
Output is correct |
19 |
Correct |
33 ms |
5284 KB |
Output is correct |
20 |
Correct |
26 ms |
5284 KB |
Output is correct |
21 |
Correct |
33 ms |
5284 KB |
Output is correct |
22 |
Correct |
19 ms |
5284 KB |
Output is correct |
23 |
Correct |
36 ms |
5284 KB |
Output is correct |
24 |
Correct |
26 ms |
5284 KB |
Output is correct |
25 |
Correct |
26 ms |
5284 KB |
Output is correct |
26 |
Correct |
0 ms |
5284 KB |
Output is correct |
27 |
Correct |
0 ms |
5284 KB |
Output is correct |
28 |
Correct |
0 ms |
5284 KB |
Output is correct |
29 |
Correct |
0 ms |
5284 KB |
Output is correct |
30 |
Correct |
0 ms |
5284 KB |
Output is correct |
31 |
Correct |
0 ms |
5284 KB |
Output is correct |
32 |
Correct |
0 ms |
5284 KB |
Output is correct |
33 |
Correct |
0 ms |
5284 KB |
Output is correct |
34 |
Correct |
0 ms |
5284 KB |
Output is correct |
35 |
Correct |
0 ms |
5284 KB |
Output is correct |
36 |
Correct |
0 ms |
5284 KB |
Output is correct |
37 |
Correct |
3 ms |
5284 KB |
Output is correct |
38 |
Correct |
0 ms |
5284 KB |
Output is correct |
39 |
Correct |
0 ms |
5284 KB |
Output is correct |
40 |
Correct |
0 ms |
5284 KB |
Output is correct |