[BOJ][🟡5][백준#02636] 치즈

문제 출처

BAEKJOON Online Judge #2636

문제

아래 <그림 1>과 같이 정사각형 칸들로 이루어진 사각형 모양의 판이 있고, 그 위에 얇은 치즈(회색으로 표시된 부분)가 놓여 있다. 판의 가장자리(<그림 1>에서 네모 칸에 X친 부분)에는 치즈가 놓여 있지 않으며 치즈에는 하나 이상의 구멍이 있을 수 있다. 이 치즈를 공기 중에 놓으면 녹게 되는데 공기와 접촉된 칸은 한 시간이 지나면 녹아 없어진다. 치즈의 구멍 속에는 공기가 없지만 구멍을 둘러싼 치즈가 녹아서 구멍이 열리면 구멍 속으로 공기가 들어가게 된다. <그림 1>의 경우, 치즈의 구멍을 둘러싼 치즈는 녹지 않고 ‘c’로 표시된 부분만 한 시간 후에 녹아 없어져서 <그림 2>와 같이 된다.

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

<그림 2> 한 시간 후의 치즈 모양

<그림 3> 두 시간 후의 치즈 모양 <그림 3>은 원래 치즈의 두 시간 후 모양을 나타내고 있으며, 남은 조각들은 한 시간이 더 지나면 모두 녹아 없어진다. 그러므로 처음 치즈가 모두 녹아 없어지는 데는 세 시간이 걸린다. <그림 3>과 같이 치즈가 녹는 과정에서 여러 조각으로 나누어 질 수도 있다. 입력으로 사각형 모양의 판의 크기와 한 조각의 치즈가 판 위에 주어졌을 때, 공기 중에서 치즈가 모두 녹아 없어지는 데 걸리는 시간과 모두 녹기 한 시간 전에 남아있는 치즈조각이 놓여 있는 칸의 개수를 구하는 프로그램을 작성하시오.

입력

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

출력

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

예제

입력

13 12
0 0 0 0 0 0 0 0 0 0 0 0
0 0 0 0 0 0 0 0 0 0 0 0
0 0 0 0 0 0 0 1 1 0 0 0
0 1 1 1 0 0 0 1 1 0 0 0
0 1 1 1 1 1 1 0 0 0 0 0
0 1 1 1 1 1 0 1 1 0 0 0
0 1 1 1 1 0 0 1 1 0 0 0
0 0 1 1 0 0 0 1 1 0 0 0
0 0 1 1 1 1 1 1 1 0 0 0
0 0 1 1 1 1 1 1 1 0 0 0
0 0 1 1 1 1 1 1 1 0 0 0
0 0 1 1 1 1 1 1 1 0 0 0
0 0 0 0 0 0 0 0 0 0 0 0

출력

3
5

My Sol

import sys
input = sys.stdin.readline


I, J = map(int, input().split())
mat = [list(map(int, input().split())) for _ in range(I)]
space = [[0]*J for _ in range(I)]
RemoveSet = set()
RemoveSet.add((0, 0))

time = 0
lastcnt = 0
while RemoveSet:
    lastcnt = len(RemoveSet)
    Q = RemoveSet
    for i, j in RemoveSet:
        mat[i][j] = 0

    RemoveSet = set()
    while Q:
        ni, nj = Q.pop()
        if space[ni][nj]: continue
        space[ni][nj] = 1

        for di, dj in ((-1,0),(1,0),(0,-1),(0,1)):
            si, sj = ni+di, nj+dj
            if not (0<=si<I and 0<=sj<J): continue
            if space[si][sj]: continue
            if mat[si][sj]:
                RemoveSet.add((si, sj))
                continue
            Q.add((si, sj))
    time += 1

print(time-1)
print(0 if time==1 else lastcnt)

최초 무조건 공백인 0행 0열부터 BFS를 이용해 공백의 모든 부분을 체크해준다. 가장자리가 모두 공백이므로 (0, 0)부터 시작하면 가장자리를 둘러 공기와 접촉하는 모든 부분을 BFS로 체크할 수 있다. 영역을 벗어난다면 continue하고, 치즈와 접촉한다면 해당 좌표를 RemoveSet에 추가한다. 이를 공백에 대한 Q를 채우는동안 계속 실시한다.

한 사이클이 돌면, 공기와 직접 접촉하는 치즈의 좌표가 RemoveSet에 저장되고, 이 각각을 모두 mat에서 0으로 체크한 뒤, 이 좌표들을 다시 Q로 만들어 space는 1로 체크하고 탐색좌표의 mat이 1, 즉 다시 치즈와 접촉한다면 접촉 좌표를 RemoveSet에 넣어준다. RemoveSet이 없는, 즉 치즈가 없어질 때까지 반복하며, RemoveSet의 크기를 매번 구해, RemoveSet이 없어지기 직전의 RemoveSet의 크기, 즉 마지막의 치즈 크기를 계산해준다.

처음부터 치즈가 없다면 time은 0이 될 것이고, (0,0)부터 시작하기 위해 RemoveSet에 (0,0)을 넣은 것이 lastcnt로 잡히므로 time-1이 0이라면 lastcnt도 0을 반환하는 예외처리를 하며 출력한다.

결과

맞았습니다!!