백준 10026

문제

적록색약은 빨간색과 초록색의 차이를 거의 느끼지 못한다. 따라서, 적록색약인 사람이 보는 그림은 아닌 사람이 보는 그림과는 좀 다를 수 있다.

크기가 N×N인 그리드의 각 칸에 R(빨강), G(초록), B(파랑) 중 하나를 색칠한 그림이 있다. 그림은 몇 개의 구역으로 나뉘어져 있는데, 구역은 같은 색으로 이루어져 있다. 또, 같은 색상이 상하좌우로 인접해 있는 경우에 두 글자는 같은 구역에 속한다. (색상의 차이를 거의 느끼지 못하는 경우도 같은 색상이라 한다)

예를 들어, 그림이 아래와 같은 경우에

RRRBB
GGBBB
BBBRR
BBRRR
RRRRR

적록색약이 아닌 사람이 봤을 때 구역의 수는 총 4개이다. (빨강 2, 파랑 1, 초록 1) 하지만, 적록색약인 사람은 구역을 3개 볼 수 있다. (빨강-초록 2, 파랑 1)

그림이 입력으로 주어졌을 때, 적록색약인 사람이 봤을 때와 아닌 사람이 봤을 때 구역의 수를 구하는 프로그램을 작성하시오.

입력

첫째 줄에 N이 주어진다. (1 ≤ N ≤ 100)

둘째 줄부터 N개 줄에는 그림이 주어진다.

코드

package baekjoon.dfs;

import java.io.BufferedReader;
import java.io.IOException;
import java.io.InputStreamReader;

public class Q10026 {

    static int N;
    static boolean[][] visited;
    static String[][] grid;
    static int[] dx = {-1, 1, 0, 0};
    static int[] dy = {0, 0, -1, 1};
    static boolean isColorBlindness;

    public static void main(String[] args) throws IOException {
        BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
        StringBuilder sb = new StringBuilder();

        N = Integer.parseInt(br.readLine());

        visited = new boolean[N][N];
        grid = new String[N][N];

        for (int i = 0; i < N; i++) {
            String s = br.readLine();
            String[] split = s.split("");
            if (N >= 0) System.arraycopy(split, 0, grid[i], 0, N);
        }

        int result = 0;

        // 1 색맹 X 탐색
        for (int i = 0; i < N; i++) {
            for (int j = 0; j < N; j++) {
                if(!visited[i][j]) result += dfs(i, j);
            }
        }

        sb.append(result).append(" "); // 6 결과값 저장

        // 7 초기화, 플래그 변경
        result = 0;
        visited = new boolean[N][N];
        isColorBlindness = true;

        // 8 색맹 O 탐색
        for (int i = 0; i < N; i++) {
            for (int j = 0; j < N; j++) {
                if(!visited[i][j]) result += dfs(i, j);
            }
        }

        // 9 누적 후 결과 출력
        sb.append(result);
        System.out.println(sb);
    }

    public static int dfs(int x, int y) {
        visited[x][y] = true; // 2 방문여부 업데이트

        String color = grid[x][y];

        for (int i = 0; i < dx.length; i++) { // 3 현재좌표를 기준으로 좌우하상 순으로 좌표 탐색
            int nextX = x + dx[i];
            int nextY = y + dy[i];

            if (nextX < 0 || nextY < 0 || nextX >= N || nextY >= N || visited[nextX][nextY]) continue; // 4 범위와 방문여부 유효성 체크
            String nextColor = grid[nextX][nextY];
            // 5 색맹여부에 따른 다음탐색 결정을 위한 조건문
            if (!isColorBlindness) {  
                if(color.equals(nextColor)) dfs(nextX, nextY);
            } else {
                if (color.equals("B")) {
                    if(color.equals(nextColor)) dfs(nextX, nextY);
                } else {
                    if(!nextColor.equals("B")) dfs(nextX, nextY);
                }
            }
        }

        return 1;
    }
}

해설

문제유형

탐색문제이다. DFS 또는 BFS로 풀이가 가능하다.

변수

N: 배열의 가로세로 크기
visited[][]: 방문여부 저장 배열
grid[][]: 색깔정보 저장 배열
dx[]: 탐색 X좌표
dy[]: 탐색 Y좌표
isColorBlindness: 색맹 여부

풀이과정

번호위치는 코드의 주석 참조.
전체적인 풀이과정은 데이터를 입력받고, 색맹일 경우와 비색맹일 경우 2번 탐색을 한다. isColorBlindness 변수를 통해 색맹/비색맹 여부를 판단하고, 탐색 내부에서 isColorBlindness 값에따라 탐색조건을 분기 시킨다.

주석

1 : 색맹이 아닐 경우(isColorBlindness=false), 방문하지 않은곳 탐색.
2 : 현재 좌표의 방문여부를 업데이트 한다.
3 : 현재 좌표를 기준으로 좌→우→하→상 순으로 탐색한다
4 : 배열 범위와 방문여부에 대한 유효성을 체크한다.
5 : 색맹여부에 따른 다음탐색에 대한 조건문 분기

• 색맹이 아닐경우
  • 다음 탐색위치 색깔과 현재 위치 색깔이 일치할경우 탐색
• 적록색맹일 경우
  • 현재색상이 파랑(B)일경우 :
    • 다음 탐색위치 색깔과 현재 위치 색깔이 일치할경우 탐색
  • 현재 색상이 파란색이 아닐경우(R or G) :
    • 적록색맹은 적색 녹색을 구분하지 못하기 때문에 파랑색 이외에는 같은 색깔로 판단하고 탐색한다.

6: 탐색의 결과값을 result 변수에 누적시키고, 그 결과를 StringBuilder에 저장한다.

7 : 색맹일 경우isColorBlindness=true 탐색을 위해 결과값과, 방문여부배열을 초기화하고 색맹여부플래그를 변경한다.
8 : 5번과정과 같이 색맹일 경우isColorBlindness=true를 조건으로 탐색한다.
9 : 6번과정에 같이 누적 후 결과를 문제의 조건에 맞게 출력한다.

다른사람의 풀이

큰 풀이과정은 비슷하였다. 변수선언을 조금 효율적으로 하였다.

• grid[][] 배열을 String 형 대신 char 형으로 선언하였다.
• visited[][] 배열을 선언하는 대신 탐색한 곳의 문자를 X로 바꿔주었다.
• visited[][] 배열을 사용하지 않기 때문에 dfs 메서드의 인자로 이전 탐색 grid[][] 배열을 넘겨주었다.