문제
사악한 암흑의 군주 이민혁은 드디어 마법 구슬을 손에 넣었고, 그 능력을 실험해보기 위해 근처의 티떱숲에 홍수를 일으키려고 한다. 이 숲에는 고슴도치가 한 마리 살고 있다. 고슴도치는 제일 친한 친구인 비버의 굴로 가능한 빨리 도망가 홍수를 피하려고 한다.
티떱숲의 지도는 R행 C열로 이루어져 있다. 비어있는 곳은 '.'로 표시되어 있고, 물이 차있는 지역은 '*', 돌은 'X'로 표시되어 있다. 비버의 굴은 'D'로, 고슴도치의 위치는 'S'로 나타내어져 있다.
매 분마다 고슴도치는 현재 있는 칸과 인접한 네 칸 중 하나로 이동할 수 있다. (위, 아래, 오른쪽, 왼쪽) 물도 매 분마다 비어있는 칸으로 확장한다. 물이 있는 칸과 인접해있는 비어있는 칸(적어도 한 변을 공유)은 물이 차게 된다. 물과 고슴도치는 돌을 통과할 수 없다. 또, 고슴도치는 물로 차있는 구역으로 이동할 수 없고, 물도 비버의 소굴로 이동할 수 없다.
티떱숲의 지도가 주어졌을 때, 고슴도치가 안전하게 비버의 굴로 이동하기 위해 필요한 최소 시간을 구하는 프로그램을 작성하시오.
고슴도치는 물이 찰 예정인 칸으로 이동할 수 없다. 즉, 다음 시간에 물이 찰 예정인 칸으로 고슴도치는 이동할 수 없다. 이동할 수 있으면 고슴도치가 물에 빠지기 때문이다.
입력
첫째 줄에 50보다 작거나 같은 자연수 R과 C가 주어진다.
다음 R개 줄에는 티떱숲의 지도가 주어지며, 문제에서 설명한 문자만 주어진다. 'D'와 'S'는 하나씩만 주어진다.
출력
첫째 줄에 고슴도치가 비버의 굴로 이동할 수 있는 가장 빠른 시간을 출력한다. 만약, 안전하게 비버의 굴로 이동할 수 없다면, "KAKTUS"를 출력한다.
풀이
출발위치와 물의 위치를 큐에 담아주고, BFS를 진행하였다. 나의 방식은 큐에 출발위치가 가장 먼저 담겨야 한다.
BFS를 진행할 때에 2가지 경우로 나눠줘야 한다. 현재 값이 '출발'인지 '물'인지.
현재 값이 '출발'에 나아갈 위치가 아직 방문하지 않은 좌표이고 진행할 수 있는 경로라면 나아갈 위치에 '출발' 값을 넣어준다.
현재 값이 '물'에 나아갈 위치가 아직 방문하지 않은 좌표이고 진행할 수 있느 경로라면 나아갈 위치에 '물'을 넣어준다.
이때 방문표시를 총 걸린 횟수로 쓸것이기 때문에 +1을 하면서 처리를 해준다.
탈출 조건은 도착지점 좌표의 값에 'S'가 들어오면 도착지점 좌표의 (방문 값 - 1) 을 반환하면 된다.
전체코드
import sys
from collections import deque
input = sys.stdin.readline
def bfs(start_x, start_y, end_x, end_y):
visit[start_x][start_y] = 1
# 시작지점을 제일 먼저 시작하기 위해 캐싱 해뒀다가 맨 앞으로 넣어줌
q.appendleft((start_x, start_y))
while q:
now_x, now_y = q.popleft()
# 도착지점에 고슴도치가 있으면 <- 이런식으로 판단해야함
# 현재 위치가 도착지점이면 <- 이렇게 판단하면 이미 고슴도치가 도착했어도 스택에 남아있는걸 처리하느라 거리저장값이 달라질 수 있음
# e.g. 이미 도착했어도 내 옆이 바로 물이 있고 해당 물을 처리할 차례이면 내 거리가 1로 바뀜
if graph[end_x][end_y] == 'S':
# 방문 처리와 거리를 동시에 쓰려고 첫 시작지점을 1로 했기 때문에 1을 빼줌
return visit[end_x][end_y] - 1
for dx, dy in direction:
nx, ny = now_x + dx, now_y + dy
if 0 <= nx < r and 0 <= ny < c:
# 고슴도치는 . 또는 D 로만 나아감
if not visit[nx][ny] and (graph[nx][ny] == '.' or graph[nx][ny] == 'D') and graph[now_x][now_y] == 'S':
graph[nx][ny] = 'S'
visit[nx][ny] = visit[now_x][now_y] + 1
q.append((nx, ny))
# 물은 . 또는 S 로만 나아감
elif (graph[nx][ny] == 'S' or graph[nx][ny] == '.') and graph[now_x][now_y] == '*':
graph[nx][ny] = '*'
visit[nx][ny] = 1
q.append((nx, ny))
return 'KAKTUS'
r, c = map(int, input().split())
graph = [list(input().rstrip()) for _ in range(r)]
visit = [[0] * c for _ in range(r)]
direction = [(-1, 0), (0, -1), (1, 0), (0, 1)]
q = deque()
start_x, start_y = 0, 0
end_x, end_y = 0, 0
for i in range(r):
for j in range(c):
if graph[i][j] == '*':
q.append((i, j))
elif graph[i][j] == 'S':
start_x, start_y = i, j
elif graph[i][j] == 'D':
end_x, end_y = i, j
print(bfs(start_x, start_y, end_x, end_y))