[백준/BOJ] gold4 - 2636번 치즈 (Python)

▶2636 - 치즈

백준 로고

▶문제

아래 <그림 1>과 같이 정사각형 칸들로 이루어진 사각형 모양의 판이 있고, 그 위에 얇은 치즈(회색으로 표시된 부분)가 놓여 있다. 판의 가장자리(<그림 1>에서 네모 칸에 X 친 부분)에는 치즈가 놓여 있지 않으며 치즈에는 하나 이상의 구멍이 있을 수 있다.

이 치즈를 공기 중에 놓으면 녹게 되는데 공기와 접촉된 칸은 한 시간이 지나면 녹아 없어진다. 치즈의 구멍 속에는 공기가 없지만 구멍을 둘러싼 치즈가 녹아서 구멍이 열리면 구멍 속으로 공기가 들어가게 된다. <그림 1>의 경우, 치즈의 구멍을 둘러싼 치즈는 녹지 않고 ‘c’로 표시된 부분만 한 시간 후에 녹아 없어져서 <그림 2>와 같이 된다.

<그림 1> 원래 치즈 모양 / <그림 2> 한 시간 후의 치즈 모양 / <그림 3> 두 시간 후의 치즈 모양

다시 한 시간 후에는 <그림 2>에서 ‘c’로 표시된 부분이 녹아 없어져서 <그림 3>과 같이 된다.

<그림 3>은 원래 치즈의 두 시간 후 모양을 나타내고 있으며, 남은 조각들은 한 시간이 더 지나면 모두 녹아 없어진다. 그러므로 처음 치즈가 모두 녹아 없어지는 데는 세 시간이 걸린다. <그림 3>과 같이 치즈가 녹는 과정에서 여러 조각으로 나누어질 수도 있다.

입력으로 사각형 모양의 판의 크기와 한 조각의 치즈가 판 위에 주어졌을 때, 공기 중에서 치즈가 모두 녹아 없어지는 데 걸리는 시간과 모두 녹기 한 시간 전에 남아있는 치즈조각이 놓여 있는 칸의 개수를 구하는 프로그램을 작성하시오.

 

▶입력

첫째 줄에는 사각형 모양 판의 세로와 가로의 길이가 양의 정수로 주어진다. 세로와 가로의 길이는 최대 100이다. 판의 각 가로줄의 모양이 윗 줄부터 차례로 둘째 줄부터 마지막 줄까지 주어진다. 치즈가 없는 칸은 0, 치즈가 있는 칸은 1로 주어지며 각 숫자 사이에는 빈칸이 하나씩 있다.

 

▶출력

첫째 줄에는 치즈가 모두 녹아서 없어지는 데 걸리는 시간을 출력하고, 둘째 줄에는 모두 녹기 한 시간 전에 남아있는 치즈조각이 놓여 있는 칸의 개수를 출력한다.

728x90

▶풀이

이 문제 같은 경우에는 bfs를 이용해서 풀었다.

모든 배열의 값이 0이 될 때까지 반복문을 돌려서 bfs()를 계속해서 호출했다.

 

다른 bfs()와 다른 점은 bfs()를 돌릴 때마다 기존의 값을 계속해서 변경해주어야 한다.

아래 코드를 보면 차이를 한눈에 볼 수 있다.

if 0 <= nx < n and 0 <= ny < m and visit[nx][ny] == 0:
	if arr[nx][ny] == 0:
		queue.append([nx, ny])
		visit[nx][ny] = 1
	elif arr[nx][ny] == 1:
		arr[nx][ny] = 0
		visit[nx][ny] = 1
		count += 1

기존의 bfs() 경우에는 굳이 if / elif로 나눌 필요가 없다.

하지만 이번 문제는 테두리에 있는 치즈는 녹아야 하기에 저렇게 코드를 구현하였다.

 

arr[nx][ny] == 0의 경우는 치즈가 아닌 부분이다.

이는 visit를 변경해 주고 queue에 추가해 주면 된다.

 

arr[nx][ny] == 1인 경우를 체크해 주는 게 중요하다.

이때는 값을 0으로 변경해주어야 한다.

이 부분이 치즈가 녹았다는 것을 표현해 준다.

그다음은 count를 세서 몇 개의 치즈가 녹았는지 체크해 주면 된다.

 

안쪽에 있는 치즈는 녹지 않는가?라는 의문이 들 수도 있다.

arr가 1일 때는 queue에 따로 추가해주지 않기에 안쪽에 치즈는 bfs로 탐색이 불가능하다.

그렇기에 안쪽에 있는 치즈는 탐색하지 않기에 괜찮다.

<정답 코드>

from collections import deque


def bfs():
    queue = deque()
    queue.append([0, 0])
    visit[0][0] = 1
    count = 0

    while queue:
        x, y = queue.popleft()
        for i in range(4):
            nx = x + dx[i]
            ny = y + dy[i]

            if 0 <= nx < n and 0 <= ny < m and visit[nx][ny] == 0:
                if arr[nx][ny] == 0:
                    queue.append([nx, ny])
                    visit[nx][ny] = 1
                elif arr[nx][ny] == 1:
                    arr[nx][ny] = 0
                    visit[nx][ny] = 1
                    count += 1
    return count


n, m = map(int, input().split())
arr = []
answer = []
time = 0

dx = [1, -1, 0, 0]
dy = [0, 0, 1, -1]

for i in range(n):
    arr.append(list(map(int, input().split())))

while True:
    time += 1
    visit = [[0 for _ in range(m)] for _ in range(n)]
    count = bfs()
    if count == 0:
        break
    answer.append(count)

print(time - 1)
print(answer[-1])