[BOJ] 2636/2638 치즈

[출처: BOJ]

문제(2636 치즈)

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

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

<그림 1> 원래 치즈 모양

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

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

<그림 3> 두 시간 후의 치즈 모양

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

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

입력

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

출력

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

2638

2636 문제랑 같은 문제지만 마지막 남은 치즈들의 개수를 출력할 필요는 없고, 각 치즈 칸에 대하여 4면 중에 2면이 공기에 닿으면 녹는다는 조건이 추가되었다.

풀이

두 문제다 로직이 같기 때문에 같이 올렸다. 다행히 문제에서는 0행 0열 n-1행 m-1열은 다 공기라고 해줬으니 그 점을 이용하기로 하였다. (0,0)칸부터 시작하여 0인 칸을 따라가면서 1을 만나면 체크해주기로 했다. 첫번 째 문제는 공기랑 닿아있기만 하면 한 시간 후에 녹기 때문에 바로 t_cheese 배열에 0으로 저장해주고, 한시간이 지난 후에 전부 한 번에 녹는 것이기 때문에 마지막에 cheese.copy를 통해 cheese배열을 초기화를 해줬다. 이런식으로 만약 1인 칸이 하나도 없으면 flag이 참으로 초기화되지 않기 때문에 cnt가 t_cnt로 초기화 되지 않으며 마지막에 남은 치즈의 개수를 출력하게 된다.

import sys
input = lambda: sys.stdin.readline().rstrip()
from collections import deque
n, m = map(int, input().split())
cnt = 0
dx = [1, -1, 0, 0]
dy = [0, 0, 1, -1]
cheese = [list(map(int, input().split()))for _ in range(n)]
hour = 0
cnt = 0
while True:
  t_cheese = cheese.copy()
  flag = False
  visited = [[False for _ in range(m)]for _ in range(n)]
  q = deque()
  q.append((0, 0))
  q.append((n-1, 0))
  q.append((0, m-1))
  q.append((n-1, m-1))
  visited[0][0] = True
  visited[n-1][0] = True
  visited[0][m-1] = True
  visited[n-1][m-1] = True
  t_cnt = 0
  while q:
    x, y = q.popleft()
    for i in range(4):
      nx = x + dx[i]
      ny = y + dy[i]
      if 0 <= nx < n and 0 <= ny < m:
        if not visited[nx][ny]:
          if cheese[nx][ny] == 0:
            q.append((nx, ny))
            visited[nx][ny] = True
          else:
            flag = True
            t_cnt += 1
            t_cheese[nx][ny] = 0
            visited[nx][ny] = True
  if not flag:
    break
  hour += 1
  cnt = t_cnt
  cheese = t_cheese.copy()
print(hour)
print(cnt)

2638문제는 면이 2개 이상 닿은 면만 녹기 때문에 cnt_matrix라는 배열을 선언한 후 bfs로 탐색할 때 만약 1을 만나면 그 cnt_matrix에서 그 칸에 하나씩 더해줘서 닿아있는 면을 카운트 해줬다. 만약 cnt_matrix[nx][ny]가 2보다 크면 녹기 때문에 t_cheese배열에 0으로 초기화해줬다.

import sys
input = lambda: sys.stdin.readline().rstrip()
from collections import deque
n, m = map(int, input().split())
dx = [1, -1, 0, 0]
dy = [0, 0, 1, -1]
cheese = [list(map(int, input().split()))for _ in range(n)]
hour = 0
while True:
  t_cheese = [[0 for _ in range(m)]for _ in range(n)]
  for i in range(n):
    for j in range(m):
      t_cheese[i][j] = cheese[i][j]
  flag = False
  visited = [[False for _ in range(m)]for _ in range(n)]
  cnt_matrix = [[0 for _ in range(m)]for _ in range(n)]
  q = deque()
  q.append((0, 0))
  visited[0][0] = True
  while q:
    x, y = q.popleft()
    for i in range(4):
      nx = x + dx[i]
      ny = y + dy[i]
      if 0 <= nx < n and 0 <= ny < m:
        if not visited[nx][ny]:
          if cheese[nx][ny] == 0:
            q.append((nx, ny))
            visited[nx][ny] = True
          else:
            flag = True
            cnt_matrix[nx][ny] += 1
            if cnt_matrix[nx][ny] >= 2:
              t_cheese[nx][ny] = 0
  if not flag:
    break
  hour += 1
  for i in range(n):
    for j in range(m):
      cheese[i][j] = t_cheese[i][j]
print(hour)