[백준 10026번] 적록색약

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

10026번: 적록색약

풀이) 

방문하지않았던 곳이면 dfs 돌려서 전부 방문처리해주고 구간값 1개 더하는 식으로 처리를 했고, 두번째 dfs에선 기본 값들 다시 초기화 시키고, G -> R로 바꿔놓고 다시 한번 전과 같은 방식으로 dfs를 돌려서 해결했다.

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

int N, ans = 0;
int visited[101][101];
int dirX[4] = {-1, 0, 1, 0};
int dirY[4] = {0, 1, 0, -1};
char map[101][101];

void changeMapAndReset(int N)
{
    memset(visited, 0, sizeof(visited));
    ans = 0;
    for (int i = 0; i < N; i++)
        for (int j = 0; j < N; j++)
            if (map[i][j] == 'G')
                map[i][j] = 'R';
}

void dfs(int x, int y, char color)
{
    visited[x][y] = 1;
    for (int i = 0; i < 4; i++)
    {
        int nx = x + dirX[i];
        int ny = y + dirY[i];
        if (nx < 0 || nx > N - 1 || ny < 0 || ny > N - 1)
            continue;
        if (!visited[nx][ny] && map[nx][ny] == color)
            dfs(nx, ny, color);
    }
}

int main()
{
    ios::sync_with_stdio(0);
    cin.tie(0); cout.tie(0);

    cin >> N;
    for (int i = 0; i < N; i++)
        for (int j = 0; j < N; j++)
            cin >> map[i][j];

    for (int i = 0; i < N; i++)
        for (int j = 0; j < N; j++)
            if (!visited[i][j])
            {
                dfs(i, j, map[i][j]);
                ans++;
            }
    cout << ans << " ";
    changeMapAndReset(N);
    for (int i = 0; i < N; i++)
        for (int j = 0; j < N; j++)
            if (!visited[i][j])
            {
                dfs(i, j, map[i][j]);
                ans++;
            }
    cout << ans;
    return (0);
}

 

 

제 코드는 절대 완벽하지 않습니다

더 좋은 풀이방법 혹은 코드에 관한 질문이 있으시면 언제든 댓글주세요!