치즈
티어 : Gold 5
시간 제한 : 1 초
메모리 제한 : 128 MB
알고리즘 분류 : 구현, 그래프 이론, 그래프 탐색, 너비 우선 탐색, 시뮬레이션
문제
아래 <그림 1>과 같이 정사각형 칸들로 이루어진 사각형 모양의 판이 있고, 그 위에 얇은 치즈(회색으로 표시된 부분)가 놓여 있다. 판의 가장자리(<그림 1>에서 네모 칸에 X친 부분)에는 치즈가 놓여 있지 않으며 치즈에는 하나 이상의 구멍이 있을 수 있다.
이 치즈를 공기 중에 놓으면 녹게 되는데 공기와 접촉된 칸은 한 시간이 지나면 녹아 없어진다. 치즈의 구멍 속에는 공기가 없지만 구멍을 둘러싼 치즈가 녹아서 구멍이 열리면 구멍 속으로 공기가 들어가게 된다. <그림 1>의 경우, 치즈의 구멍을 둘러싼 치즈는 녹지 않고 ‘c’로 표시된 부분만 한 시간 후에 녹아 없어져서 <그림 2>와 같이 된다.
다시 한 시간 후에는 <그림 2>에서 ‘c’로 표시된 부분이 녹아 없어져서 <그림 3>과 같이 된다.
<그림 3>은 원래 치즈의 두 시간 후 모양을 나타내고 있으며, 남은 조각들은 한 시간이 더 지나면 모두 녹아 없어진다. 그러므로 처음 치즈가 모두 녹아 없어지는 데는 세 시간이 걸린다. <그림 3>과 같이 치즈가 녹는 과정에서 여러 조각으로 나누어 질 수도 있다.
입력으로 사각형 모양의 판의 크기와 한 조각의 치즈가 판 위에 주어졌을 때, 공기 중에서 치즈가 모두 녹아 없어지는 데 걸리는 시간과 모두 녹기 한 시간 전에 남아있는 치즈조각이 놓여 있는 칸의 개수를 구하는 프로그램을 작성하시오.
입력
첫째 줄에는 사각형 모양 판의 세로와 가로의 길이가 양의 정수로 주어진다. 세로와 가로의 길이는 최대 100이다. 판의 각 가로줄의 모양이 윗 줄부터 차례로 둘째 줄부터 마지막 줄까지 주어진다. 치즈가 없는 칸은 0, 치즈가 있는 칸은 1로 주어지며 각 숫자 사이에는 빈칸이 하나씩 있다.
출력
첫째 줄에는 치즈가 모두 녹아서 없어지는 데 걸리는 시간을 출력하고, 둘째 줄에는 모두 녹기 한 시간 전에 남아있는 치즈조각이 놓여 있는 칸의 개수를 출력한다.
예제 입출력
Algorithm
DFS - 재귀, 구현
1. 1시간마다 DFS를 이용해 치즈에 구멍이 있는지 확인
☞ 0으로된 덩어리가 2개 이상인 경우 구멍 존재
2. 구멍이 존재하면 구멍에 해당하는 칸의 값을 H로 변경
3. graph를 한 번씩 돌면서 현재 위치가 1인데 상, 하, 좌, 우 중 한 칸이라도 0이 존재하면 이번에 삭제되는 칸
☞ 해당 칸의 값을 C로 변경
4. 매 번 C의 개수와 1의 개수의 차를 확인해 0인 경우 반복문 break
Code
def dfs(x, y, replace_value):
# graph 밖으로 벗어나면 False Return
if x < 0 or x > N-1 or y < 0 or y > M-1:
return False
# 이미 방문했던 칸이거나 0이 아니면 False Return
if visited[x][y] or graph[x][y] != 0:
return False
# 현재 위치 방문 기록
visited[x][y] = True
# 구멍인 경우 H로 변경
graph[x][y] = replace_value
# 상하좌우로 이동
dfs(x-1, y, replace_value)
dfs(x+1, y, replace_value)
dfs(x, y-1, replace_value)
dfs(x, y+1, replace_value)
import sys
input = sys.stdin.readline
N, M = map(int, input().split())
sys.setrecursionlimit(N*M)
graph = []
count_cheezes = 0 # 치즈가 녹기 전 치즈조각이 놓여 있는 칸의 개수
for _ in range(N):
graph.append(list(map(int, input().split())))
count_cheezes += graph[-1].count(1)
num = 0
while True:
# 구멍의 값을 H로 변경
visited = [[False for _ in range(M)] for _ in range(N)]
for x in range(N):
for y in range(M):
# x, y가 0, 0이 아니면 replace 값 H로 설정
if x > 0 and y > 0:
dfs(x, y, 'H')
else:
dfs(x, y, 0)
# 가장자리에 있는 1을 C로 변경
count_c = 0
for x in range(N):
for y in range(M):
if graph[x][y] == 1:
# 현재 index의 상하좌우 중 한 군데라도 0이 있으면 가장자리
for dx, dy in [(-1, 0), (1, 0), (0, -1), (0, 1)]:
nx = x+dx
ny = y+dy
# graph 밖으로 벗어나면 continue
if nx < 0 or nx > N-1 or ny < 0 or ny > M-1:
continue
# nx, ny가 0이면 x, y를 C로 변경하고 break
if graph[nx][ny] == 0:
graph[x][y] = 'C'
count_c += 1
break
# 현재 1인 칸과 없어질 칸의 차가 0이면 break
if count_cheezes - count_c == 0:
print(num+1)
print(count_cheezes)
break
# C, H인 칸 모두 0으로 변경
for x in range(N):
for y in range(M):
if graph[x][y] == 'C'or graph[x][y] == 'H':
graph[x][y] = 0
num += 1
count_cheezes -= count_c메모리: 33988 KB
시간: 148 ms
'백준 > Python' 카테고리의 다른 글
| [백준 10845] 큐 Python (0) | 2022.04.20 |
|---|---|
| [백준 20056] 마법사 상어와 파이어볼 Python (0) | 2022.04.20 |
| [백준 1743] 음식물 피하기 Python (0) | 2022.04.15 |
| [백준 18429] 근손실 Python (0) | 2022.04.15 |
| [백준 1316] 그룹 단어 체커 Python (0) | 2022.04.15 |
