[Baekjoon Online Judge] 10026 - 적록색약 / Java

https://www.acmicpc.net/problem/10026

10026번: 적록색약

import java.io.BufferedReader;
import java.io.FileInputStream;
import java.io.InputStreamReader;

public class Main {

	// 상 하 좌 우
	static int[] dr = { -1, 1, 0, 0 };
	static int[] dc = { 0, 0, -1, 1 };
	static int N;
	static char[][] colorMap; // RGB
	static boolean[][] visited; // 방문 체크

	public static void main(String[] args) throws Exception {
		BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
		N = Integer.parseInt(br.readLine());

		colorMap = new char[N][N];
		for (int i = 0; i < N; i++) {
			colorMap[i] = br.readLine().toCharArray();
		}

		// 적록색약이 아닌 사람
		int cntIsNotRGColorblind = 0; // 적록색약 아닌 사람이 보는 구역
		visited = new boolean[N][N];
		for (int i = 0; i < N; i++) {
			for (int j = 0; j < N; j++) {
				if (!visited[i][j]) {
					check(i, j, false); // false : 적록색약아닌 사람
					cntIsNotRGColorblind++;
				}
			}
		}

		// 적록색약인 사람
		int cntIsRGColorblind = 0; // 적록색약인 사람이 보는 구역
		visited = new boolean[N][N];
		for (int i = 0; i < N; i++) {
			for (int j = 0; j < N; j++) {
				if (!visited[i][j]) {
					check(i, j, true); // true : 적록색약인 사람
					cntIsRGColorblind++;
				}
			}
		}

		System.out.println(cntIsNotRGColorblind + " " + cntIsRGColorblind);
	}

	private static void check(int r, int c, boolean isRGColorblind) {
		visited[r][c] = true; // 현재 좌표 방문 체크

		for (int dir = 0; dir < 4; dir++) {
			int nr = r + dr[dir];
			int nc = c + dc[dir];

			// 경계 벗어나거나 방문한 곳이면 다음 방향 탐색
			if (nr < 0 || nr >= N || nc < 0 || nc >= N || visited[nr][nc]) {
				continue;
			}

			char currentColor = colorMap[r][c];
			char nextColor = colorMap[nr][nc];

			if (currentColor == nextColor // 현재 색과 다음 색이 같은지
					// 적록색약인 사람이면 R, G 같은 것으로 보기
					|| (isRGColorblind && (currentColor == 'R' || currentColor == 'G')
										&& (nextColor == 'R' || nextColor == 'G'))) {
				// 다음 좌표를 시작으로 같은 동작 수행
                check(nr, nc, isRGColorblind);
			}
		}
	}
}