[BFS] 10026번 적록색약

10026번 적록색약

 

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

 

문제

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

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

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

RRRBB
GGBBB
BBBRR
BBRRR
RRRRR

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

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

 

입력

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

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

 

출력

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

 

예제 입력

5
RRRBB
GGBBB
BBBRR
BBRRR
RRRRR

 

예제 출력

4 3

 

해결방법

구역의 개수를 구하는 간단한 BFS 문제이다

DFS 탐색 또는 BFS 탐색 중 하나를 선택해 문제를 접근하면 되는데 이 문제에서는 BFS로 접근했다

 

단 적록색약 O vs 적록색약 X 두 가지 경우의 BFS 탐색을 모두 진행해야한다

• 배열의 값을 변경하는 방법
• BFS 함수를 2개 만드는 방법
• Flag를 받아서 조건문으로 해결하는 방법

이렇게 3가지 방법이 존재했는데 Flag를 받아서 해결하였다

 

// 적록색약인 사람
if(dyschromatopsia){
    if(map[r][c] == 'R' || map[r][c] == 'G'){
        if(!chk[nx][ny] && (map[nx][ny] == 'R' || map[nx][ny] == 'G')){
            q.push(pii(nx,ny));
            chk[nx][ny] = true;
        }
    }else{
        if(!chk[nx][ny] && map[nx][ny] == 'B'){
            q.push(pii(nx,ny));
            chk[nx][ny] = true;
        }
    }
}

// 적록색약이 아닌 사람
else{
    if(!chk[nx][ny] && map[nx][ny] == map[r][c]){
        q.push(pii(nx,ny));
        chk[nx][ny] = true;
    }
}

 

 

소스코드

#include <iostream>
#include <stdio.h>
#include <math.h>
#include <cstring>
#include <vector>
#include <queue>
#include <algorithm>
#define MAX 101
#define INF 987654321
using namespace std;

typedef pair<int, int> pii;

int n,ans1,ans2;

char map[MAX][MAX];
bool chk[MAX][MAX];

int dx[4] = {0,1,0,-1};
int dy[4] = {1,0,-1,0};

void bfs(int r,int c,bool dyschromatopsia){
    queue<pii> q;
    q.push(pii(r,c));
    chk[r][c] = true;
    
    while(!q.empty()){
        int x = q.front().first;
        int y = q.front().second;
        q.pop();
        
        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) continue;
            
            // 적록색약인 사람
            if(dyschromatopsia){
                if(map[r][c] == 'R' || map[r][c] == 'G'){
                    if(!chk[nx][ny] && (map[nx][ny] == 'R' || map[nx][ny] == 'G')){
                        q.push(pii(nx,ny));
                        chk[nx][ny] = true;
                    }
                }else{
                    if(!chk[nx][ny] && map[nx][ny] == 'B'){
                        q.push(pii(nx,ny));
                        chk[nx][ny] = true;
                    }
                }
            }
            // 적록색약이 아닌 사람
            else{
                if(!chk[nx][ny] && map[nx][ny] == map[r][c]){
                    q.push(pii(nx,ny));
                    chk[nx][ny] = true;
                }
            }
        }
    }
}

int main(int argc, const char * argv[]) {
    // cin,cout 속도향상
    ios_base::sync_with_stdio(false);
    cin.tie(NULL); cout.tie(NULL);
    
    cin >> n;
    for(int i=0; i<n; i++){
        for(int j=0; j<n; j++){
            cin >> map[i][j];
        }
    }
    
    // 적록색약이 아닌 사람의 경우
    memset(chk, false, sizeof(chk));
    ans1 = 0;
    for(int i=0; i<n; i++){
        for(int j=0; j<n; j++){
            if(!chk[i][j]){
                bfs(i,j,false);
                ans1++;
            }
        }
    }
    
    // 적록색약인 사람의 경우
    memset(chk, false, sizeof(chk));
    ans2 = 0;
    for(int i=0; i<n; i++){
        for(int j=0; j<n; j++){
            if(!chk[i][j]){
                bfs(i,j,true);
                ans2++;
            }
        }
    }
    
    cout << ans1 << " " << ans2 << endl;
}