[BaekJoon] 10026 적록색약 :: monton

문제

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

풀이

dfs를 이용한 간단한 문제이다

나는 조건을 너무 더럽게 짠거 같다

코드

#include <iostream>
 
using namespace std;
 
int N,cnt1,cnt2;
char arr[101][101];
int visit[2][101][101];
int dir[4][2] = { {-1,0},{0,1},{1,0},{0,-1} };
 
void dfs(int x, int y,char c) {
    if (x < 0 || x >= N || y < 0 || y >= N) return;
    if (visit[0][x][y]|| arr[x][y] != c) return;
    visit[0][x][y] = 1;
 
    for (int i = 0; i < 4; i++) {
            dfs(x + dir[i][0], y + dir[i][1], c);
    }
}
 
void dfs2(int x, int y, char c) {
    if (x < 0 || x >= N || y < 0 || y >= N) return;
    if (visit[1][x][y]) return;
    if (c == 'R') {
        if (arr[x][y] != 'G'&& arr[x][y] != 'R') return;
    }
    else if (c == 'G') {
        if (arr[x][y] != 'R'&& arr[x][y] != 'G') return;
    }
    else if (c != arr[x][y]) return;
 
    visit[1][x][y] = 1;
 
    for (int i = 0; i < 4; i++) {
        dfs2(x + dir[i][0], y + dir[i][1], c);
    }
}
 
int main() {
    cin >> N;
 
    for (int i = 0; i < N; i++) {
        for (int j = 0; j < N; j++) {
            cin >> arr[i][j];
        }
    }
    
    for (int i = 0; i < N; i++) {
        for (int j = 0; j < N; j++) {
            if (!visit[0][i][j]) ++cnt1,dfs(i,j,arr[i][j]);
            if (!visit[1][i][j]) ++cnt2,dfs2(i,j,arr[i][j]);
        }
    }
    cout << cnt1 << " " << cnt2;
}