[백준/10026] 적록색약 - JAVA

이지환·2025년 4월 22일

백준 알고리즘

알고리즘(백준) 💻

목록 보기

58/80

📌 문제

알고리즘 분류 : 그래프
난이도 : 골드5
출처 : 백준 - 적록색약

🦧 문제 풀이 접근

DFS방식으로 해결했다.
일반적인 경우와 적녹색약인 경우를 각각 나눠서 구현했다.

💻 code

import java.util.*;
import java.io.*;
public class Main {
    public static void main(String[] args) throws IOException {
        BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
        int N = Integer.parseInt(br.readLine());
        char picture[][] = new char[N][N];
        boolean[][] check1 = new boolean[N][N], check2 = new boolean[N][N];
        for(int i=0;i<N;i++) {
            String line = br.readLine();
            for(int j=0;j<N;j++) {
                picture[i][j] = line.charAt(j);
            }
        }
        int cnt1=0, cnt2=0;
        for(int i=0;i<N;i++) {
            for(int j=0;j<N;j++) {
                if(!check1[i][j]) {
                    check1[i][j]=true;
                    DFS1(picture, check1, i, j);
                    cnt1++;
                }
                if(!check2[i][j]) {
                    check2[i][j]=true;
                    DFS2(picture, check2, i, j);
                    cnt2++;
                }
            }
        }
        System.out.println(cnt1 +" "+cnt2);
    }

    static void DFS2(char[][] picture, boolean[][] check2, int i, int j) {
        int[] di = {0,1,0,-1}, dj = {1,0,-1,0};
        for(int k=0;k<4;k++) {
            int ci = i+di[k], cj = j+dj[k];
            if(ci<0 || picture.length-1<ci || cj<0 || picture[0].length-1<cj)
                continue;
            if(!check2[ci][cj]) {
                if(picture[i][j]=='B' && picture[ci][cj]=='B') {
                    check2[ci][cj]=true;
                    DFS2(picture,check2,ci,cj);
                }
                else if(picture[i][j]!='B' && picture[ci][cj]!='B') {
                    check2[ci][cj] = true;
                    DFS2(picture, check2, ci, cj);
                }
            }
        }
    }

    static void DFS1(char[][] picture, boolean[][] check1, int i, int j) {
        int[] di = {0,1,0,-1}, dj = {1,0,-1,0};
        for(int k=0;k<4;k++) {
            int ci = i+di[k], cj = j+dj[k];
            if(ci<0 || picture.length-1<ci || cj<0 || picture[0].length-1<cj)
                continue;
            if(picture[i][j] == picture[ci][cj] && !check1[ci][cj]) {
                check1[ci][cj]=true;
                DFS1(picture,check1,ci,cj);
            }
        }
    }
}