BOJ 10026 (적록색약)

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

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

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

RRRBB
GGBBB
BBBRR
BBRRR
RRRRR

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

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

• 입력
  첫째 줄에 N이 주어진다. (1 ≤ N ≤ 100)
  둘째 줄부터 N개 줄에는 그림이 주어진다.
  
  
• 출력
  적록색약이 아닌 사람이 봤을 때의 구역의 개수와 적록색약인 사람이 봤을 때의 구역의 수를 공백으로 구분해 출력한다.

#include<iostream>
using namespace std;
int N;
char picture[101][101];
bool visited[101][101];
int dy[4] = { -1,1,0,0 };
int dx[4] = { 0,0,-1,1 };
int nc, ac;
void fast_io()
{
	ios_base::sync_with_stdio(false);
	cin.tie(NULL);
}

void dfs(int y, int x)
{
	for (int i = 0; i < 4; i++)
	{
		int ny = y + dy[i];
		int nx = x + dx[i];
		if (ny < 0 || ny >= N || nx < 0 || nx >= N)
		{
			continue;
		}
		if (!visited[ny][nx] && picture[ny][nx] == picture[y][x]) //이전 깊이와의 색이 같은지
		{
			visited[ny][nx] = true;
			dfs(ny, nx);
		}
	}
}

int main()
{
	void fast_io();
	cin >> N;
	for (int i = 0; i < N; i++)
	{
		for (int j = 0; j < N; j++)
		{
			cin >> picture[i][j];
		}
	}
	for (int i = 0; i < N; i++)
	{
		for (int j = 0; j < N; j++)
		{
			if (!visited[i][j])
			{
				nc++;
				visited[i][j] = true;
				dfs(i, j);
			}
		}
	}
	for (int i = 0; i < N; i++)
	{
		for (int j = 0; j < N; j++)
		{
			visited[i][j] = false;
			if (picture[i][j] == 'G')
			{
				picture[i][j] = 'R';
			}
		}
	}
	for (int i = 0; i < N; i++)
	{
		for (int j = 0; j < N; j++)
		{
			if (!visited[i][j])
			{
				ac++;
				visited[i][j] = true;
				dfs(i, j);
			}
		}
	}
	cout << nc << ' ' << ac;
}

  깊이우선탐색(DFS)를 이용하면 생각보다 쉬운 문제이다. DFS 함수에서 이전 위치와의 색이 같은지 점검해주는 작업이 핵심 + 빨간색 초록색 구별이 어려운 사람의 영역을 확인할 때는 그냥 두 색을 같게 만들어 버리면 된다.