적록색약이 아닌 사람이 봤을 때 구역의 갯수, 적록색약인 사람이 봤을 때 구역의 갯수를 순서대로 출력합니다.
적록색약이 아닌 사람 먼저 bfs 순회를 돌아서 구역의 갯수를 구하고, 빨간색과 초록색 중 하나를 다른 하나의 색으로 모두 바꾼 후에 다시 bfs를 돌리면 적록색약인 사람이 봤을 때 구역의 갯수를 구할 수 있습니다.
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
|
import java.io.BufferedReader;
import java.io.InputStreamReader;
import java.util.LinkedList;
import java.util.Queue;
import java.util.StringTokenizer;
public class Main {
static BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
static StringTokenizer st;
static boolean visit[][];
static int direct[][] = { { 0, 1 }, { 1, 0 }, { 0, -1 }, { -1, 0 } };
static char list[][];
static int N;
static void bfs(int sx, int sy) {
Queue<int[]> q = new LinkedList<>();
q.add(new int[] { sx, sy });
visit[sx][sy] = true;
while (!q.isEmpty()) {
int x = q.peek()[0];
int y = q.poll()[1];
for (int i = 0; i < 4; i++) {
int nx = x + direct[i][0];
int ny = y + direct[i][1];
if (nx < 0 || ny < 0 || nx >= N || ny >= N)
continue;
if (visit[nx][ny] || list[x][y] != list[nx][ny])
continue;
q.add(new int[] { nx, ny });
visit[nx][ny] = true;
}
}
}
static void func() {
int ansFirst = 0;
int ansSecond = 0;
for (int i = 0; i < N; i++) {
for (int j = 0; j < N; j++) {
if (visit[i][j])
continue;
bfs(i, j);
ansFirst++;
}
}
for (int i = 0; i < N; i++) {
for (int j = 0; j < N; j++) {
visit[i][j] = false;
if (list[i][j] == 'G')
list[i][j] = 'R';
}
}
for (int i = 0; i < N; i++) {
for (int j = 0; j < N; j++) {
if (visit[i][j])
continue;
bfs(i, j);
ansSecond++;
}
}
System.out.println(ansFirst + " " + ansSecond);
}
static void input() throws Exception {
st = new StringTokenizer(br.readLine());
N = Integer.parseInt(st.nextToken());
list = new char[N][N];
visit = new boolean[N][N];
for (int i = 0; i < N; i++) {
st = new StringTokenizer(br.readLine());
list[i] = st.nextToken().toCharArray();
}
}
public static void main(String[] args) throws Exception {
input();
func();
}
}
|
cs |
'algorithm > bfs' 카테고리의 다른 글
boj 17135 캐슬 디펜스 (0) | 2021.02.17 |
---|---|
boj 16236 아기 상어 (0) | 2021.02.14 |
boj 2206 벽 부수고 이동하기 (0) | 2021.02.13 |
boj 14502 연구소 (0) | 2021.02.12 |
boj 20304 비밀번호 제작 (0) | 2021.02.08 |