[백준] 10026번 적록색약 - 파이썬

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

10026번: 적록색약

풀이

문제는 BFS로 풀어주었습니다. 정상인 사람과 적록색약 인 사람이 구분할 수 있는 영역의 개수를 출력하면 되는 문제입니다.

적록색약인 사람은 빨간색(R)과 녹색(G)이 구분되지 않기에 한가지의 색으로만 보입니다. 그래서 적록색약의 사람이 보는 그림의 색 중에서 R을 G로 바꾸어주었습니다. 문자열 메서드 중 하나인 replace() 메서드를 사용해주었습니다.

그 이후에는 BFS를 진행해주어서 한 영역을 탐색완료할 때마다 구역 개수를 추가해주었습니다

# 10026번 적록색약
import sys
from collections import deque

def bfs(graph):
    ans = 0
    visited = [[False] * n for _ in range(n)]
    for i in range(n):
        for j in range(n):
            if visited[i][j] == False:
                ans += 1
                queue = deque()
                queue.append((j,i))
                visited[i][j] = True
                color = graph[i][j]
                while queue:
                    x, y= queue.popleft()
                    for a in range(4):
                        nx = x + dx[a]
                        ny = y + dy[a]
                        if 0 <= nx < n and 0 <= ny < n:
                            if visited[ny][nx] == False and graph[ny][nx] == color:
                                visited[ny][nx] = True
                                queue.append((nx, ny))
    return ans

# 좌표 이동 값
dx = [0, 0, 1, -1]
dy = [1, -1, 0, 0]
n = int(input())
# 정상인이 보는 그림
graph_normal = [sys.stdin.readline().rstrip() for _ in range(n)]
# 적록색약 사람이 보는 그림
graph_red_green = [i.replace("R","G") for i in graph_normal]  # 문자열 replace함수를 사용 

normal = bfs(graph_normal)
weak_eyes = bfs(graph_red_green)

print(f'{normal} {weak_eyes}')

출력결과