https://www.acmicpc.net/problem/10026
10026번: 적록색약
문제 적록색약은 빨간색과 초록색의 차이를 거의 느끼지 못한다. 따라서, 적록색약인 사람이 보는 그림은 아닌 사람이 보는 그림과는 좀 다를 수 있다. 크기가 N×N인 그리드의 각 칸에 R(빨강), G(초록), B(파랑) 중 하나를 색칠한 그림이 있다. 그림은 몇 개의 구역으로 나뉘어져 있는데, 구역은 같은 색으로 이루어져 있다. 또, 같은 색상이 상하좌우로 인접해 있는 경우에 두 글자는 같은 구역에 속한다. (색상의 차이를 거의 느끼지 못하는 경우도 같은
www.acmicpc.net
| 목표: 적록색약인 사람과 아닌 사람이 봤을 때의 구역의 수 구하기 2차원 배열, 4방 탐색 DFS 활용 적록색약인 사람은 G과 R이 구별이 없기 때문에 조건을 주어서 2차원 배열에 G가 있으면 R로 변경 |
import java.util.Arrays;
import java.util.Scanner;
public class Main {
static int N, p = 0, np = 0;
static char[][] map;
static boolean[][] visited;
static int[][] dir = { { 0, 1 }, { 1, 0 }, { -1, 0 }, { 0, -1 } };
public static void main(String[] args) {
Scanner sc = new Scanner(System.in);
N = sc.nextInt();
map = new char[N][N];
visited = new boolean[N][N];
for (int i = 0; i < N; i++) {
String s = sc.next();
for (int j = 0; j < N; j++) {
map[i][j] = s.charAt(j);
}
}
for (int i = 0; i < N; i++) {
for (int j = 0; j < N; j++) {
if (map[i][j] == 'R' && !visited[i][j]) {
DFS(i, j);
np++;
} else if (map[i][j] == 'B' && !visited[i][j]) {
DFS(i, j);
np++;
} else if (map[i][j] == 'G' && !visited[i][j]) {
DFS(i, j);
np++;
}
}
}
System.out.print(np+" ");
visited = new boolean[N][N];
for (int i = 0; i < N; i++) {
for (int j = 0; j < N; j++) {
if (map[i][j] == 'G') {
map[i][j] = 'R';
}
}
}
for (int i = 0; i < N; i++) {
for (int j = 0; j < N; j++) {
if (map[i][j] == 'R' && !visited[i][j]) {
DFS(i, j);
p++;
}
else if (map[i][j] == 'B' && !visited[i][j]) {
DFS(i, j);
p++;
}
}
}
System.out.println(p);
}// end of Main
public static void DFS(int x, int y) {
visited[x][y] = true;
for (int i = 0; i < 4; i++) {
int nx = x + dir[i][0];
int ny = y + dir[i][1];
if (isInside(nx, ny) && map[nx][ny] == map[x][y] && !visited[nx][ny]) {
DFS(nx, ny);
}
}
}
public static boolean isInside(int x, int y) {
return x >= 0 && x < N && y >= 0 && y < N;
}
}// end of class'ALGORITHM > 프로그래머스 | 백준 | 삼성 | 카카오' 카테고리의 다른 글
| [프로그래머스] 주식가격 (0) | 2020.05.04 |
|---|---|
| [백준] 13335번 트럭 (0) | 2020.04.02 |
| [백준] 3985번 롤케이크 (0) | 2020.04.01 |
| [백준] 2577번 숫자의 개수 (0) | 2020.04.01 |
| [백준] 2999번 비밀 이메일 (0) | 2020.04.01 |