📩 출처
문제
사악한 암흑의 군주 이민혁은 드디어 마법 구슬을 손에 넣었고, 그 능력을 실험해보기 위해 근처의 티떱숲에 홍수를 일으키려고 한다. 이 숲에는 고슴도치가 한 마리 살고 있다. 고슴도치는 제일 친한 친구인 비버의 굴로 가능한 빨리 도망가 홍수를 피하려고 한다.
티떱숲의 지도는 R행 C열로 이루어져 있다. 비어있는 곳은 '.'로 표시되어 있고, 물이 차있는 지역은 '*', 돌은 'X'로 표시되어 있다. 비버의 굴은 'D'로, 고슴도치의 위치는 'S'로 나타내어져 있다.
매 분마다 고슴도치는 현재 있는 칸과 인접한 네 칸 중 하나로 이동할 수 있다. (위, 아래, 오른쪽, 왼쪽) 물도 매 분마다 비어있는 칸으로 확장한다. 물이 있는 칸과 인접해있는 비어있는 칸(적어도 한 변을 공유)은 물이 차게 된다. 물과 고슴도치는 돌을 통과할 수 없다. 또, 고슴도치는 물로 차있는 구역으로 이동할 수 없고, 물도 비버의 소굴로 이동할 수 없다.
티떱숲의 지도가 주어졌을 때, 고슴도치가 안전하게 비버의 굴로 이동하기 위해 필요한 최소 시간을 구하는 프로그램을 작성하시오.
고슴도치는 물이 찰 예정인 칸으로 이동할 수 없다. 즉, 다음 시간에 물이 찰 예정인 칸으로 고슴도치는 이동할 수 없다. 이동할 수 있으면 고슴도치가 물에 빠지기 때문이다.
입력
첫째 줄에 50보다 작거나 같은 자연수 R과 C가 주어진다.
다음 R개 줄에는 티떱숲의 지도가 주어지며, 문제에서 설명한 문자만 주어진다. 'D'와 'S'는 하나씩만 주어진다.
출력
첫째 줄에 고슴도치가 비버의 굴로 이동할 수 있는 가장 빠른 시간을 출력한다. 만약, 안전하게 비버의 굴로 이동할 수 없다면, "KAKTUS"를 출력한다.
👉 생각
- 너비우선 탐색을 통해서 비버집 까지 가는 최단거리를 구하도록 접근했다.
- 고슴도치의 좌표와 물이 있는 곳의 좌표들을 담아서 bfs 함수의 인자로 넣어준다.
- 각각 q와 water라는 배열을 생성해서 담고 방문처리를 체크할 이차원 배열도 만들어 준다.
- 고슴도치가 움직일 수 있는 방향을 체크하기 전에 water에 있는 모든 물들이 갈 수 있는 곳을 먼저 물로 채워준다.
- 여기서 cnt를 사용했는데, cnt는 시간을 의미하고 1시간이 지났을 때 고슴도치가 갈 수 있는 칸을 가기 전에 물이 1시간에 갈 수 있는 모든 좌표들을 물로 채워 준다. 그러면 고슴도치가 갈 수 있는 곳이 제한된다.
- cnt가 2일 때는 고슴도치가 2시간이 되었을 때 갈 수 있는 곳인데, 그 전에 물들이 2시간이 되었을 때 갈 수 있는 모든 곳을 물로 채워 준다. 그러면 고슴도치가 갈 수 있는 곳이 또 제한된다.
- 위와 같은 방법을 반복해서 비버 집을 만나면 cnt를 리턴하고 만나지 못하면 KAKTUS를 반환한다!
from collections import deque
def bfs(x, y, water):
q = deque([])
q.append((x, y, 0))
visited = [[0] * c for _ in range(r)]
visited[x][y] =1
visited2 = [[0] * c for _ in range(r)]
for w1, w2, tmp in water:
visited2[w1][w2] = 1
while q:
x, y, cnt = q.popleft()
if arr[x][y] == 'D':
return cnt
if water:
while water[0][-1] == cnt:
w1, w2, time = water.pop(0)
for dw1, dw2 in [(-1,0), (1,0), (0,-1), (0,1)]:
nw1, nw2 = w1 + dw1, w2 + dw2
if 0 <= nw1 < r and 0 <= nw2 < c and arr[nw1][nw2] != 'D' and arr[nw1][nw2] != 'X' and not visited2[nw1][nw2]:
arr[nw1][nw2] = '*'
visited2[nw1][nw2] = 1
water.append((nw1, nw2, time+1))
if not water:
break
for dx, dy in [(-1,0), (1,0), (0,-1), (0,1)]:
nx, ny = x + dx, y + dy
if 0 <= nx < r and 0 <= ny < c and arr[nx][ny] != '*' and arr[nx][ny] != 'X' and not visited[nx][ny]:
q.append((nx, ny, cnt+1))
visited[nx][ny] = 1
return 'KAKTUS'
r, c = map(int, input().split())
arr = [list(input()) for _ in range(r)]
water = []
w1, w2 = -100, - 100
for i in range(r):
for j in range(c):
if arr[i][j] == '*':
water.append((i, j, 0))
if arr[i][j] == 'S':
s1, s2 = i, j
print(bfs(s1, s2, water))
