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

10026번: 적록색약

BFS 연습 #1

 

전형적인 BFS 탐색 문제이다

색약인 경우에 'R' 과 'G' 를 같다 생각하고 탐색하게 해주면 문제없이 풀 수 있다 

#include <iostream>
#include <string>
#include <queue>
#include <utility>
 
#define pii pair<int, int>
 
using namespace std;
 
int N;
char Map[101][101];
bool visited[101][101][2];
int dr[4] = {0, 0, 1, -1}, dc[4] = {1, -1, 0, 0};
 
void input(){
    string in;
    cin >> N;
    for(int i = 0; i < N; i++){
        cin >> in;
        for(int j = 0; j < N; j++){
            Map[i][j] = in[j];
        }
    }
}
 
void BFS(int r, int c, int d, char color){
    queue<pii> que;
    que.push({r, c});
    visited[r][c][d] = true;
    
    while(!que.empty()){
        pii now = que.front();
        que.pop();
        
        for(int i = 0; i < 4; i++){
            int nr = now.first + dr[i];
            int nc = now.second + dc[i];
            if(nr < 0 || nr >= N || nc < 0 || nc >= N || visited[nr][nc][d]) continue;
            if(d == 0){     //색약이 아닌경우
                if(Map[nr][nc] == color){
                    visited[nr][nc][d] = true;
                    que.push({nr, nc});
                }
            }
            else{   // 색약인 경우
                if(color == Map[nr][nc]){
                    visited[nr][nc][d] = true;
                    que.push({nr, nc});
                }
                else if((color == 'R' && Map[nr][nc] == 'G') || (color == 'G' && Map[nr][nc] == 'R')){
                    visited[nr][nc][d] = true;
                    que.push({nr, nc});
                }
            }
        }
    }
}
 
void solve(){
    int ncount = 0, cwcount = 0;
    for(int i = 0; i < N; i++){
        for(int j = 0; j < N; j++){
            char nowColor = Map[i][j];
            if(!visited[i][j][0]){
                BFS(i, j, 0, nowColor);
                ncount++;
            }
            if(!visited[i][j][1]){
                BFS(i, j, 1, nowColor);
                cwcount++;
            }
        }
    }
    cout << ncount << " " << cwcount;
}
 
int main(){
    input();
    solve();
    
    return 0;
}
 

BFS