[백준 10026: 적록색약] - C++

www.acmicpc.net/problem/10026

10026번: 적록색약

#include<iostream>
#include<string>
#include<string.h>
using namespace std;

int n;
char map[101][101];
bool check[101][101];

int dir[4][2] = {
	{-1,0}
	,{0,1}
	,{1,0}
	,{0,-1}
};

void D2(int _x, int _y, char curChar)
{

	for (int i = 0; i < 4; i++)
	{
		int nextX = _x + dir[i][0];
		int nextY = _y + dir[i][1];

		if (nextX < 0 || nextY < 0 || n <= nextX || n <= nextY)
			continue;


		if (!check[nextX][nextY])
		{
			
			if (curChar == 'R'&& (map[nextX][nextY] == 'R' || map[nextX][nextY] == 'G'))
			{
				check[nextX][nextY] = true;
				D2(nextX, nextY,'G');
				D2(nextX, nextY, 'R');
			}
			else if (curChar == 'G'&& (map[nextX][nextY] == 'R' || map[nextX][nextY] == 'G'))
			{				
				check[nextX][nextY] = true;
				D2(nextX, nextY, 'R');
				D2(nextX, nextY, 'G');
			}
			else if (map[nextX][nextY] == curChar)
			{
				check[nextX][nextY] = true;
				D2(nextX, nextY, curChar);
				
			}
		}
	}
}

void D(int _x, int _y,char curChar)
{

	for (int i = 0; i < 4; i++)
	{
		int nextX = _x + dir[i][0];
		int nextY = _y + dir[i][1];

		if (nextX < 0 || nextY < 0 || n <= nextX || n <= nextY)
			continue;
		

		if (map[nextX][nextY]==curChar && !check[nextX][nextY])
		{
			check[nextX][nextY] = true;
			D(nextX, nextY,curChar);			
		}
	}
}

int main() {
	
	cin >> n;

	string input;
	for (int i = 0; i < n; i++)
	{
		cin >> input;
		for (int j = 0; j < n; j++)
		{
			map[i][j] = input[j];
		}
	}
	int cnt = 0;

	for (int i = 0; i < n; i++)
	{
		for (int j = 0; j < n; j++)
		{
			if (check[i][j] == false)
			{
				char curChar = map[i][j];
				check[i][j] = true;
				D(i, j, curChar);
				cnt++;
			}
			
		}
	}

	cout << cnt << " ";

	cnt = 0;
	memset(check, false, sizeof(check));
	for (int i = 0; i < n; i++)
	{
		for (int j = 0; j < n; j++)
		{
			if (check[i][j] == false)
			{
				char curChar = map[i][j];

				check[i][j] = true;
				D2(i, j, curChar);
				cnt++;
			}

		}
	}

	cout << cnt << endl;
	return 0;
}