14-janghw0126

[janghw0126]

🔗 문제 링크

백준 | 적록색약

✔️ 소요된 시간

1시간

✨ 수도 코드

문제를 봤을 때, 적록색약인 사람을 봤을 때와 아닌 사람이 봤을 때 구역의 수를 구해야 하므로 빨강, 초록, 파랑 각각의 색을 모두 탐색해야 된다는 것을 알 수 있습니다. 따라서, 각 색상별로 재귀를 돌려야하기 때문에 재귀 깊이가 깊을 수 있으므로 적록색약의 특성에 따라 빨강, 초록색을 구분할 수 없는 케이스를 만들어줘야 합니다.

Note

핵심 로직

1. 정상인과 적록색맹을 구분하여 각자의 그룹을 셉니다.
2. BFS를 사용해 그룹을 탐색합니다.
3. 정상인인 경우, 한번에 하나의 그룹을 탐색하고 그 그룹을 모두 방문한 후 그룹의 개수를 하나 증가시킵니다.
4. 적록색맹인 경우, 빨간색과 초록색을 구분하지 않기 때문에, 적록색맹에서는 초록색을 빨간색으로 바꿔서 하나의 색깔로 취급하고 다시 그룹을 센 후 연결된 점들을 계속 탐색합니다.

✨최종 코드

import sys
from collections import deque

input = sys.stdin.readline

def bfs(x, y, graph, visited):
    q = deque()
    q.append((x, y))
    visited[x][y] = True
    
    # 상하좌우 방향 선언하기
    dx, dy = [1, -1, 0, 0], [0, 0, 1, -1]
    
    while q:
        x, y = q.popleft()
        
        for i in range(4):
            nx, ny = x + dx[i], y + dy[i]
            
            if 0 <= nx < n and 0 <= ny < n and not visited[nx][ny]:
                # 색상이 같고 방문하지 않은 경우에만 큐에 넣음
                if graph[x][y] == graph[nx][ny]:
                    visited[nx][ny] = True
                    q.append((nx, ny))

n = int(input())
graph = [list(input().strip()) for _ in range(n)]

# 첫 번째 방문 배열은 정상인 (R, G, B 구분함)
visited = [[False] * n for _ in range(n)]
normal_count = 0

# 두 번째 방문 배열은 적록색맹인 (R, G 구분 안 함)
visited_rg = [[False] * n for _ in range(n)]
rg_count = 0

# 정상인 그룹 개수 구하기
for i in range(n):
    for j in range(n):
        if not visited[i][j]:
            bfs(i, j, graph, visited)
            normal_count += 1

# 적록색맹인 경우 'G'를 'R'로 바꾸기
for i in range(n):
    for j in range(n):
        if graph[i][j] == 'G':
            graph[i][j] = 'R'

# 적록색맹인 그룹 개수 구하기
for i in range(n):
    for j in range(n):
        if not visited_rg[i][j]:
            bfs(i, j, graph, visited_rg)
            rg_count += 1

print(normal_count, rg_count)

📚 새롭게 알게된 내용

.

[jung0115]

코드는 좀 지저분 하지만... 저도 비슷하게 풀었습니다 👍
핵심 로직을 간결하게 설명해주셔서 더 이해가 쉬웠습니다! 이번 차시도 수고하셨어요 ☺️

import java.io.InputStreamReader;
import java.io.BufferedReader;
import java.io.IOException;
import java.util.LinkedList;
import java.util.Queue;

public class Main {
  static int[] dx = {-1, 1, 0, 0};
  static int[] dy = {0, 0, -1, 1};
  
  static int N;
  static Character[][] weakMap;
  static Character[][] strongMap;
  static boolean[][] weakVisited;
  static boolean[][] strongVisited;
  static int weakCount = 0, strongCount = 0;
    
  public static void main(String[] args) throws IOException {
    BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
    N = Integer.parseInt(br.readLine());

    strongMap = new Character[N][N];
    weakMap = new Character[N][N];
    
    // 맵 정보 입력
    for (int i = 0; i < N; i++) {
      String str = br.readLine();
      for (int j = 0; j < N; j++) {
        strongMap[i][j] = str.charAt(j);
        weakMap[i][j] = (str.charAt(j) == 'G') ? 'R' : str.charAt(j); // 약한 색맹 처리
      }
    }

    weakVisited = new boolean[N][N];
    strongVisited = new boolean[N][N];

    // 두 맵 모두에 대해 BFS 실행
    for (int i = 0; i < N; i++) {
      for (int j = 0; j < N; j++) {
        if (!weakVisited[i][j]) {
          bfs(i, j, weakMap, weakVisited);
          weakCount++;
        }
        if (!strongVisited[i][j]) {
          bfs(i, j, strongMap, strongVisited);
          strongCount++;
        }
      }
    }

    // 결과 출력
    System.out.print(strongCount + " " + weakCount);
  }

  // BFS 탐색 함수
  static void bfs(int x, int y, Character[][] map, boolean[][] visited) {
    Queue<int[]> queue = new LinkedList<>();
    queue.add(new int[]{x, y});
    visited[x][y] = true;
    Character color = map[x][y];

    while (!queue.isEmpty()) {
      int[] pos = queue.poll();
      int curX = pos[0];
      int curY = pos[1];

      // 네 방향 탐색
      for (int i = 0; i < 4; i++) {
        int moveX = curX + dx[i];
        int moveY = curY + dy[i];

        if (moveX >= 0 && moveX < N && moveY >= 0 && moveY < N && !visited[moveX][moveY] && map[moveX][moveY] == color) {
          visited[moveX][moveY] = true;
          queue.add(new int[]{moveX, moveY});
        }
      }
    }
  }
}