문제

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

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

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

RRRBB
GGBBB
BBBRR
BBRRR
RRRRR

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

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

입력

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

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

출력

적록색약이 아닌 사람이 봤을 때의 구역의 개수와 적록색약인 사람이 봤을 때의 구역의 수를 공백으로 구분해 출력한다.

입출력 예

풀이

1. 정상인 - 같은 색이 나오는 부분을 bfs로 탐색
2. 모든 요소를 다 탐색할 때까지 bfs가 실행된 횟수가 정상인이 본 구역
3. 적록색약 - R를 G로 바꾸고 1~2과정 실행

코드

java

import java.io.BufferedReader;
import java.io.IOException;
import java.io.InputStreamReader;
import java.util.LinkedList;
import java.util.Queue;

public class Main {
	static char map[][];
	static int n;
	static boolean[][] visited;
	static int[]dx = {1,-1,0,0};
	static int[]dy = {0,0,1,-1};
	static void bfs(int x,int y){
		visited[x][y]=true;
		Queue<int[]> q = new LinkedList<int[]>();
		q.offer(new int[] {x,y});
		while(!q.isEmpty()){
			int[] temp = q.poll();
			x=temp[0];
			y=temp[1];
			for(int i=0;i<4;i++) {
				int nx=x+dx[i];
				int ny=y+dy[i];
				if(nx>=0&&nx<n&&ny>=0&&ny<n&&!visited[nx][ny]&&map[x][y]==map[nx][ny]) {
					visited[nx][ny]=true;
					q.offer(new int[]{nx,ny});
				}
			}
		}
	}
	public static void main(String[] args) throws IOException {
		BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
		n = Integer.parseInt(br.readLine());
		map=new char[n][n];
		for(int i=0;i<n;i++) 
			map[i]=br.readLine().toCharArray();
		visited=new boolean[n][n];
		int answer=0;
		for(int i=0;i<n;i++) {
			for(int j=0;j<n;j++) {
				if(!visited[i][j]) {
					bfs(i,j);
					answer++;
				}
			}
		}
		System.out.print(answer+" ");
		answer=0;
		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(!visited[i][j]) {
					bfs(i,j);
					answer++;
				}
			}
		}
		System.out.print(answer);
	}
}

Python

from collections import deque
import sys
input = sys.stdin.readline
def bfs(graph,visited,i,j):
    queue=deque([[i,j]])
    while queue:
        x,y=queue.popleft()
        if visited[x][y]==0:
            visited[x][y]=1
            for k in range(4):
                dx=x+d[k][0]
                dy=y+d[k][1]
                if 0<=dx<n and 0<=dy<n and visited[dx][dy]==0 and graph[x][y]==graph[dx][dy]:
                    queue.append([dx,dy])
n=int(input())
graph=[list(input()) for _ in range(n)]
count1=0
count2=0
d=[(1,0),(-1,0),(0,1),(0,-1)]
visited=[[0]*n for _ in range(n)]
for i in range(n):
    for j in range(n):
        if visited[i][j]==0:
            bfs(graph,visited,i,j)
            count1+=1
for i in range(n):
    for j in range(n):
        if graph[i][j]=="G":graph[i][j]="R"
visited=[[0]*n for _ in range(n)]           
for i in range(n):
    for j in range(n):
        if visited[i][j]==0:
            bfs(graph,visited,i,j)
            count2+=1
print(count1,count2)