[Baekjoon] 10026. 적록색약

문제 설명

문제

적록색약은 빨간색과 초록색의 차이를 거의 느끼지 못한다. 따라서, 적록색약인 사람이 보는 그림은 아닌 사람이 보는 그림과는 좀 다를 수 있다.

크기가 N×N인 그리드의 각 칸에 R(빨강), G(초록), B(파랑) 중 하나를 색칠한 그림이 있다. 그림은 몇 개의 구역으로 나뉘어져 있는데, 구역은 같은 색으로 이루어져 있다. 또, 같은 색상이 상하좌우로 인접해 있는 경우에 두 글자는 같은 구역에 속한다. (색상의 차이를 거의 느끼지 못하는 경우도 같은 색상이라 한다)

예를 들어, 그림이 아래와 같은 경우에

RRRBB
GGBBB
BBBRR
BBRRR
RRRRR

적록색약이 아닌 사람이 봤을 때 구역의 수는 총 4개이다. (빨강 2, 파랑 1, 초록 1) 하지만, 적록색약인 사람은 구역을 3개 볼 수 있다. (빨강-초록 2, 파랑 1)

그림이 입력으로 주어졌을 때, 적록색약인 사람이 봤을 때와 아닌 사람이 봤을 때 구역의 수를 구하는 프로그램을 작성하시오.

입력

첫째 줄에 N이 주어진다. (1 ≤ N ≤ 100)

둘째 줄부터 N개 줄에는 그림이 주어진다.

출력

적록색약이 아닌 사람이 봤을 때의 구역의 개수와 적록색약인 사람이 봤을 때의 구역의 수를 공백으로 구분해 출력한다.

예제 입력 1

5
RRRBB
GGBBB
BBBRR
BBRRR
RRRRR

예제 출력 1

4 3

출처

Olympiad > USA Computing Olympiad > 2013-2014 Season > USACO March 2014 Contest > Bronze 3번

• 문제를 번역한 사람: baekjoon
• 어색한 표현을 찾은 사람: corea

알고리즘 분류

• 그래프 이론
• 그래프 탐색
• 너비 우선 탐색
• 깊이 우선 탐색

문제 풀이

# DFSBFS

DFS/BFS를 이용하는 문제입니다.

풀이 과정

전형적인 BFS 문제입니다. 땅따먹기처럼, BFS를 통해 전체 구역이 몇 개의 하위 영역으로 이루어져 있는지 구하면 됩니다. 이런 문제 유형은 상당히 자주 나오는 편입니다.

전체 코드

전체 코드입니다.

from collections import deque
didj = [(-1,0), (1,0), (0,-1), (0,1)]
def bfs(cur, state):
    if visited[state][cur[0]][cur[1]]:
        return 0
    q = deque([cur])
    visited[state][cur[0]][cur[1]] = True
    c = MAP[cur[0]][cur[1]]
    if state == 'RG/B' and c in 'RG': c = 'RG'
    while q:
        cur = q.popleft()
        for di, dj in didj:
            new_i, new_j = cur[0]+di, cur[1]+dj
            if 0<=new_i<N and 0<=new_j<N and MAP[new_i][new_j] in c and not visited[state][new_i][new_j]:
                q.append((new_i, new_j))
                visited[state][new_i][new_j] = True
    return 1

N = int(input())
MAP = []
for _ in range(N):
    MAP.append(list(input()))
visited = {'R/G/B':[[False for _ in range(N)] for _ in range(N)], 'RG/B':[[False for _ in range(N)] for _ in range(N)]}
section = {'R/G/B':0, 'RG/B':0}
for i in range(N):
    for j in range(N):
        for state in ['R/G/B', 'RG/B']:
            section[state] += bfs((i,j),state)
print(*section.values())

정리