📩 출처
문제
N×M의 모눈종이 위에 아주 얇은 치즈가 <그림 1>과 같이 표시되어 있다. 단, N 은 세로 격자의 수이고, M 은 가로 격자의 수이다. 이 치즈는 냉동 보관을 해야만 하는데 실내온도에 내어놓으면 공기와 접촉하여 천천히 녹는다. 그런데 이러한 모눈종이 모양의 치즈에서 각 치즈 격자(작 은 정사각형 모양)의 4변 중에서 적어도 2변 이상이 실내온도의 공기와 접촉한 것은 정확히 한시간만에 녹아 없어져 버린다. 따라서 아래 <그림 1> 모양과 같은 치즈(회색으로 표시된 부분)라면 C로 표시된 모든 치즈 격자는 한 시간 후에 사라진다.
<그림 2>와 같이 치즈 내부에 있는 공간은 치즈 외부 공기와 접촉하지 않는 것으로 가정한다. 그러므 로 이 공간에 접촉한 치즈 격자는 녹지 않고 C로 표시된 치즈 격자만 사라진다. 그러나 한 시간 후, 이 공간으로 외부공기가 유입되면 <그림 3>에서와 같이 C로 표시된 치즈 격자들이 사라지게 된다.
모눈종이의 맨 가장자리에는 치즈가 놓이지 않는 것으로 가정한다. 입력으로 주어진 치즈가 모두 녹아 없어지는데 걸리는 정확한 시간을 구하는 프로그램을 작성하시오.
입력
첫째 줄에는 모눈종이의 크기를 나타내는 두 개의 정수 N, M (5 ≤ N, M ≤ 100)이 주어진다. 그 다음 N개의 줄에는 모눈종이 위의 격자에 치즈가 있는 부분은 1로 표시되고, 치즈가 없는 부분은 0으로 표시된다. 또한, 각 0과 1은 하나의 공백으로 분리되어 있다.
출력
출력으로는 주어진 치즈가 모두 녹아 없어지는데 걸리는 정확한 시간을 정수로 첫 줄에 출력한다.
👉 생각
- 처음에는 치즈 (1)인 곳을 찾고 치즈의 4방향 중 비어 있는 곳이 2군데 이상인 곳을 찾도록 하는 bfs 함수를 만들었다. 치즈 주위의 비어있는 공간이 외부와 접촉하는지 안하는지 여부를 너비우선탐색을 통해 x나 y좌표가 가장자리에 속하다면 외부와 접촉하는 칸이라고 여기고 1을 반환하도록 했다.
- 즉 각 치즈 주위의 4방향이 인덱스 범위를 만족한다면 bfs 함수의 인자로 들어가고 외부 공기와 차단이 된다면 0을 외부 공기와 접촉한다면 1을 반환하여 합이 2 이상이면 해당 시간에 사라지는 치즈라고 생각을 하고 접근했다.
- pypy로 통과했지만 아무래도 for문과 if 문이 많다보니 시간초과가 발생했다. 아무래도 매번 치즈를 찾는 부분이 문제였던 것 같다.
from collections import deque
import sys
def bfs(x, y):
q = deque([])
q.append((x, y))
visited = [[False]*m for _ in range(n)]
visited[x][y] = True
while q:
x, y = q.popleft()
# 뚫려있음
if x == 0 or x == n-1 or y == 0 or y == m-1:
return 1
for dx, dy in [(-1,0), (1,0), (0,-1), (0,1)]:
nx, ny = x + dx, y + dy
if 0 <= nx < n and 0 <= ny < m and not cheese[nx][ny] and not visited[nx][ny]:
q.append((nx, ny))
visited[nx][ny] = True
return 0
n, m = map(int, input().split())
cheese = [list(map(int, sys.stdin.readline().split())) for _ in range(n)]
cnt = 0
while True:
lst = []
for i in range(n):
for j in range(m):
if cheese[i][j] == 1:
tmp = 0
for di, dj in [(-1,0), (1,0), (0,-1), (0,1)]:
ni, nj = i + di, j + dj
if 0 <= ni < n and 0 <= nj < m and cheese[ni][nj] == 0:
tmp += bfs(ni, nj)
if tmp >= 2:
lst.append((i,j))
break
for x, y in lst:
cheese[x][y] = 0
cnt += 1
check = 0
for ch in cheese:
check += sum(ch)
if check == 0:
print(cnt)
break- 다른 방법으로는 0,0부터 시작해서 치즈를 찾으면 치즈의 값에 1을 더하고 0을 만나면 방문처리만 해준다. 이렇게 되면 2변 이상이 접촉한다는 것은 치즈의 값이 3 이상이 되고 이는 해당 시간에 사라지는 치즈를 뜻한다.
- 위 방법을 치즈가 없을 때 까지 즉, 0 보다 큰 값이 없을 때까지 반복한다.
import sys
from collections import deque
def bfs(x, y):
q = deque([])
q.append((0,0))
visited = [[0] * m for _ in range(n)]
visited[0][0] = 1
while q:
x, y = q.popleft()
for dx, dy in [(-1, 0), (1, 0), (0, -1), (0, 1)]:
nx, ny = x + dx, y + dy
if 0 <= nx < n and 0 <= ny < m and not visited[nx][ny]:
if cheese[nx][ny]:
cheese[nx][ny] += 1
else:
visited[nx][ny] = 1
q.append((nx,ny))
n, m = map(int, input().split())
cheese = [list(map(int, sys.stdin.readline().split())) for _ in range(n)]
cnt = 0
while True:
bfs(0, 0)
check = True
for i in range(n):
for j in range(m):
if cheese[i][j] >= 3:
cheese[i][j] = 0
elif cheese[i][j] > 0:
cheese[i][j] = 1
check = False
cnt += 1
if check:
print(cnt)
break