[백준 10026번] 적록색약

문제 링크:https://www.acmicpc.net/problem/10026

10026번: 적록색약

<문제 풀이>

정상인에 대해서 BFS를 돌리고, 적록색약에 대해서 BFS를 돌리면 되는데

적록색약은 빨간색과 초록색을 같은 색으로 보니깐 board[i][j]의 값이 R이면 G로 바꿔준 뒤 BFS를 돌리면 됩니다.

 

<C++ 소스 코드>

#include <iostream>
#include <string>
#include <algorithm>
#include <queue>
#include <vector>
#include <utility>
using namespace std;
 
#define X first
#define Y second
 
char board[100][100];
bool visited[100][100];
int dx[4] = { 1, 0, -1, 0 };
int dy[4] = { 0, -1, 0, 1 };
int n;
int bfs() {
    int cnt = 0;
    for (int i = 0; i < n; i++) {
        for (int j = 0; j < n; j++) {
            if (!visited[i][j]) {
                cnt++;
                queue<pair<int, int>> Q;
                Q.push({ i,j });
                while (!Q.empty()) {
                    auto cur = Q.front(); Q.pop();
                    for (int dir = 0; dir < 4; dir++) {
                        int nx = cur.X + dx[dir];
                        int ny = cur.Y + dy[dir];
                        if (nx < 0 || ny < 0 || nx >= n || ny >= n)continue;
                        if (board[cur.X][cur.Y] != board[nx][ny] || visited[nx][ny]) continue; //다른 색이면 무시
                        Q.push({ nx, ny });
                        visited[nx][ny] = true;
 
                    }
                }
            }
        }
    }
    return cnt;
}
//적록 색약은 빨간색과 초록색이 같은색
void change_R_to_G() {
    for (int i = 0; i < n; i++) {
        for (int j = 0; j < n; j++) {
            if (board[i][j] == 'R') board[i][j] = 'G';
        }
    }
}
int main(void) {
    ios::sync_with_stdio(false);
    cin.tie(nullptr);
    cin >> n;
    for (int i = 0; i < n; i++) {
        string s; cin >> s;
        int s_len = s.size();
        for (int j = 0; j < s_len; j++) {
            board[i][j] = s[j];
        }
    }
    cout << bfs() << ' ';
    change_R_to_G();
    fill(&visited[0][0], &visited[99][100], 0);
    cout << bfs();
    return 0;
}