[백준/파이썬] 10026번 풀이
업데이트:
문제 정보
- 문제 출처: 백준 온라인 저지
- 문제 링크: 10026번 문제
- 문제풀이 코드 GitHub 링크
- 제출 언어: Python 3
풀이
문제
적록색약은 빨간색과 초록색의 차이를 거의 느끼지 못한다. 따라서, 적록색약인 사람이 보는 그림은 아닌 사람이 보는 그림과는 좀 다를 수 있다.
크기가 N×N인 그리드의 각 칸에 R(빨강), G(초록), B(파랑) 중 하나를 색칠한 그림이 있다. 그림은 몇 개의 구역으로 나뉘어져 있는데, 구역은 같은 색으로 이루어져 있다. 또, 같은 색상이 상하좌우로 인접해 있는 경우에 두 글자는 같은 구역에 속한다. (색상의 차이를 거의 느끼지 못하는 경우도 같은 색상이라 한다)
예를 들어, 그림이 아래와 같은 경우에
RRRBB GGBBB BBBRR BBRRR RRRRR
적록색약이 아닌 사람이 봤을 때 구역의 수는 총 4개이다. (빨강 2, 파랑 1, 초록 1) 하지만, 적록색약인 사람은 구역을 3개 볼 수 있다. (빨강-초록 2, 파랑 1)
그림이 입력으로 주어졌을 때, 적록색약인 사람이 봤을 때와 아닌 사람이 봤을 때 구역의 수를 구하는 프로그램을 작성하시오.
입력 요약
첫째 줄에 N이 주어진다. (1 ≤ N ≤ 100)
둘째 줄부터 N개 줄에는 그림이 주어진다.
출력 요약
적록색약이 아닌 사람이 봤을 때의 구역의 개수와 적록색약인 사람이 봤을 때의 구역의 수를 공백으로 구분해 출력한다.
코드
import sys;read=sys.stdin.readline
import collections
n = int(read())
paint = [list(read()[:-1]) for _ in range(n)]
q = collections.deque([])
enQ = q.append
deQ = q.popleft
def bfs():
global n, paint
area = cnt = 0
visited = [[False]*n for _ in range(n)]
while cnt < n*n:
for i in range(n):
for j in range(n):
if not visited[i][j]:
visited[i][j] = True
enQ((i,j))
cnt += 1
break
if q: break
while q:
i,j = deQ()
color = paint[i][j]
if 0<i and not visited[i-1][j] and paint[i-1][j] == color:
visited[i-1][j] = True
enQ((i-1,j))
cnt += 1
if 0<j and not visited[i][j-1] and paint[i][j-1] == color:
visited[i][j-1] = True
enQ((i,j-1))
cnt += 1
if n-1>i and not visited[i+1][j] and paint[i+1][j] == color:
visited[i+1][j] = True
enQ((i+1,j))
cnt += 1
if n-1>j and not visited[i][j+1] and paint[i][j+1] == color:
visited[i][j+1] = True
enQ((i,j+1))
cnt += 1
area += 1
return area
a=bfs()
for i in range(n):
for j in range(n):
if paint[i][j] == 'R': paint[i][j] = 'G'
print(a, bfs())
설명
핵심은 구현 관점에서 적록색약은 빨간색과 초록색의 차이를 거의 느끼지 못한다. 따라서, 적록색약인 사람이 보는 그림은 아닌 사람이 보는 그림과는 좀 다를 수 있다.
크기가 N×N인 그리드의 각 칸에 R(빨강), G(초록), B(파랑) 중 하나를 색칠한 그림이 있다. …를 만족하도록 로직을 구성하는 것입니다.
코드는 입력을 파싱한 뒤 조건 분기와 계산을 순서대로 수행하고, 문제에서 요구한 형식으로 결과를 출력합니다.
경계값과 예외 케이스도 함께 고려해 오답이 나기 쉬운 상황을 방지합니다.
댓글남기기