아래 <그림 1>과 같이 정사각형 칸들로 이루어진 사각형 모양의 판이 있고, 그 위에 얇은 치즈(회색으로 표시된 부분)가 놓여 있다. 판의 가장자리(<그림 1>에서 네모 칸에 X친 부분)에는 치즈가 놓여 있지 않으며 치즈에는 하나 이상의 구멍이 있을 수 있다.
이 치즈를 공기 중에 놓으면 녹게 되는데 공기와 접촉된 칸은 한 시간이 지나면 녹아 없어진다. 치즈의 구멍 속에는 공기가 없지만 구멍을 둘러싼 치즈가 녹아서 구멍이 열리면 구멍 속으로 공기가 들어가게 된다. <그림 1>의 경우, 치즈의 구멍을 둘러싼 치즈는 녹지 않고 ‘c’로 표시된 부분만 한 시간 후에 녹아 없어져서 <그림 2>와 같이 된다.
다시 한 시간 후에는 <그림 2>에서 ‘c’로 표시된 부분이 녹아 없어져서 <그림 3>과 같이 된다.
<그림 3>은 원래 치즈의 두 시간 후 모양을 나타내고 있으며, 남은 조각들은 한 시간이 더 지나면 모두 녹아 없어진다. 그러므로 처음 치즈가 모두 녹아 없어지는 데는 세 시간이 걸린다. <그림 3>과 같이 치즈가 녹는 과정에서 여러 조각으로 나누어 질 수도 있다.
입력으로 사각형 모양의 판의 크기와 한 조각의 치즈가 판 위에 주어졌을 때, 공기 중에서 치즈가 모두 녹아 없어지는 데 걸리는 시간과 모두 녹기 한 시간 전에 남아있는 치즈조각이 놓여 있는 칸의 개수를 구하는 프로그램을 작성하시오.
첫째 줄에는 사각형 모양 판의 세로와 가로의 길이가 양의 정수로 주어진다. 세로와 가로의 길이는 최대 100이다. 판의 각 가로줄의 모양이 윗 줄부터 차례로 둘째 줄부터 마지막 줄까지 주어진다. 치즈가 없는 칸은 0, 치즈가 있는 칸은 1로 주어지며 각 숫자 사이에는 빈칸이 하나씩 있다.
출력
첫째 줄에는 치즈가 모두 녹아서 없어지는 데 걸리는 시간을 출력하고, 둘째 줄에는 모두 녹기 한 시간 전에 남아있는 치즈조각이 놓여 있는 칸의 개수를 출력한다.
dfs : 입력받은 q와 검사할 value, 그리고 주어진 visited set에 대해 DFS를 수행한다. 최종적으로 visited set이 업데이트된다.
initialize : 초기화 함수. (0, 0)점에 대해 dfs를 수행하여 공기 좌표인 air_set을 업데이트하고, 맵 전체를 탐색하며 남은 치즈의 개수를 구하여 반환한다.
is_adj_air : cheese_list[y][x] == 1 인 cheese에 대해, 인접 좌표가 air_set에 포함되는지에 대한 여부를 반환한다.
check_cheese : 모든 cheese_list[y][x] == 1 인 cheese에 대해, 인접 좌표가 air_set에 포함되는지 여부를 조사하고 만약 포함된다면 removed_list에 담아 반환한다.
fill : removed_list에 담긴 모든 좌표의 cheese_list값을 0으로 변경하고, air_set을 큐로 하여 dfs를 수행. 공기와 접촉되는 새로운 영역들을 air_set에 포함시킨다.
solve : 메인함수. initialize 함수를 통해 초기 치즈 면적값 left를 구하고, left가 0이 될 때가지 반복문을 순회한다. check_cheese로 이번 시간에 제거될 치즈 리스트 removed_list를 구하고, fill을 통해 전체 cheese_list및 air_set을 업데이트한다. 마지막으로 시간과 left값을 업데이트한다. 만약 반복문을 탈출한다면, 현재 누적 시간과 left 및 직전의 removed_list의 길이를 이용하여 문제에서 요구하는 값들을 출력한다.
풀이 코드
dx = [-1, 1, 0, 0]
dy = [0, 0, -1, 1]
air_set = set()
N, M = map(int, input().split())
cheese_list = [list(map(int, input().split())) for _ in range(N)]
def dfs(q, val, visited_set):
_q = list(q)
while _q :
x, y = _q.pop()
for k in range(4) :
ax, ay = x+dx[k], y+dy[k]
if -1 < ax < M and -1 < ay < N and cheese_list[ay][ax] == val and (ax, ay) not in visited_set :
visited_set.add((ax, ay))
_q.append((ax, ay))
def initialize():
left = 0
dfs([(0, 0)], 0, air_set)
for i in range(N) :
for j in range(M) :
if cheese_list[i][j] == 1 :
left += 1
return left
def is_adj_air(x, y) :
for k in range(4) :
ax, ay = x + dx[k], y + dy[k]
if -1 < ax < M and -1 < ay < N and cheese_list[ay][ax] == 0 and (ax, ay) in air_set :
return True
return False
def check_cheese() :
removed_list = list()
for i in range(N) :
for j in range(M) :
if cheese_list[i][j] == 1 and is_adj_air(j, i) :
removed_list.append((j, i))
return removed_list
def fill_air(removed_list) :
for x, y in removed_list :
cheese_list[y][x] = 0
dfs(air_set, 0, air_set)
def solve() :
t = 0
left = initialize()
while left :
removed_list = check_cheese()
fill_air(removed_list)
left -= len(removed_list)
t += 1
print(t)
print(left + len(removed_list))
solve()
