1 x 1 크기의 칸들로 이루어진 직사각형 격자 형태의 미로에서 탈출하려고 합니다. 각 칸은 통로 또는 벽으로 구성되어 있으며, 벽으로 된 칸은 지나갈 수 없고 통로로 된 칸으로만 이동할 수 있습니다. 통로들 중 한 칸에는 미로를 빠져나가는 문이 있는데, 이 문은 레버를 당겨서만 열 수 있습니다. 레버 또한 통로들 중 한 칸에 있습니다. 따라서, 출발 지점에서 먼저 레버가 있는 칸으로 이동하여 레버를 당긴 후 미로를 빠져나가는 문이 있는 칸으로 이동하면 됩니다. 이때 아직 레버를 당기지 않았더라도 출구가 있는 칸을 지나갈 수 있습니다. 미로에서 한 칸을 이동하는데 1초가 걸린다고 할 때, 최대한 빠르게 미로를 빠져나가는데 걸리는 시간을 구하려 합니다.
미로를 나타낸 문자열 배열 maps가 매개변수로 주어질 때, 미로를 탈출하는데 필요한 최소 시간을 return 하는 solution 함수를 완성해주세요. 만약, 탈출할 수 없다면 -1을 return 해주세요.
제한사항
5 ≤ maps의 길이 ≤ 100
5 ≤ maps[i]의 길이 ≤ 100
maps[i]는 다음 5개의 문자들로만 이루어져 있습니다.
S : 시작 지점
E : 출구
L : 레버
O : 통로
X : 벽
시작 지점과 출구, 레버는 항상 다른 곳에 존재하며 한 개씩만 존재합니다.
출구는 레버가 당겨지지 않아도 지나갈 수 있으며, 모든 통로, 출구, 레버, 시작점은 여러 번 지나갈 수 있습니다.
입출력 예mapsresult
["SOOOL","XXXXO","OOOOO","OXXXX","OOOOE"]
16
["LOOXS","OOOOX","OOOOO","OOOOO","EOOOO"]
-1
입출력 예 설명
입출력 예 #1
주어진 문자열은 다음과 같은 미로이며
다음과 같이 이동하면 가장 빠른 시간에 탈출할 수 있습니다.
4번 이동하여 레버를 당기고 출구까지 이동하면 총 16초의 시간이 걸립니다. 따라서 16을 반환합니다.
입출력 예 #2
주어진 문자열은 다음과 같은 미로입니다.
시작 지점에서 이동할 수 있는 공간이 없어서 탈출할 수 없습니다. 따라서 -1을 반환합니다.
내 풀이
from collections import deque
#S -> L, L -> E
def solution(maps):
answer = -1
N = len(maps)
M = len(maps[0])
dx = [-1, 1, 0, 0]
dy = [0, 0, -1, 1]
def cnt_dists(start, end):
answer = 0
dists = [[10000 for _ in range(M)] for _ in range(N)]
end_reached = False
queue = deque()
for i in range(N):
for j in range(M):
if maps[i][j] == start:
queue.append([i, j])
dists[i][j] = 0
while queue:
x, y = queue.popleft()
if maps[x][y] == end:
return dists[x][y]
for i in range(4):
nx = x + dx[i]
ny = y + dy[i]
if 0 <= nx < N and 0 <= ny < M and maps[nx][ny] != 'X':
if dists[nx][ny] == 10000:
queue.append([nx, ny])
dists[nx][ny] = dists[x][y] + 1
to_lever = cnt_dists('S', 'L')
to_end = cnt_dists('L', 'E')
if not to_lever or not to_end:
return -1
return to_lever + to_end
시작점에서 레버로, 레버에서 종점으로 두번의 길이를 구해서 더하는게 핵심이었다.
두번의 길이를 따로 구해야 한번 지나갔던 길을 여러번 지나가게 할 수 있다.
그래서 cnt_dists라는 함수를 정의하여 재사용할 수 있도록 했다.
처음엔 'O'인 부분만 지나갈 수 있게 해서 실패가 떴다. 이를 꼭 주의해야한다. 레버나 시작점, 종점도 그냥 지나갈 수 있다.
