문제

1 x 1 크기의 칸들로 이루어진 직사각형 격자 형태의 미로에서 탈출하려고 합니다. 각 칸은 통로 또는 벽으로 구성되어 있으며, 벽으로 된 칸은 지나갈 수 없고 통로로 된 칸으로만 이동할 수 있습니다. 통로들 중 한 칸에는 미로를 빠져나가는 문이 있는데, 이 문은 레버를 당겨서만 열 수 있습니다. 레버 또한 통로들 중 한 칸에 있습니다. 따라서, 출발 지점에서 먼저 레버가 있는 칸으로 이동하여 레버를 당긴 후 미로를 빠져나가는 문이 있는 칸으로 이동하면 됩니다. 이때 아직 레버를 당기지 않았더라도 출구가 있는 칸을 지나갈 수 있습니다. 미로에서 한 칸을 이동하는데 1초가 걸린다고 할 때, 최대한 빠르게 미로를 빠져나가는데 걸리는 시간을 구하려 합니다.

미로를 나타낸 문자열 배열 maps가 매개변수로 주어질 때, 미로를 탈출하는데 필요한 최소 시간을 return 하는 solution 함수를 완성해주세요. 만약, 탈출할 수 없다면 -1을 return 해주세요.

제한 사항

5 ≤ maps의 길이 ≤ 100
5 ≤ maps[i]의 길이 ≤ 100

풀이

아이디어

• BFS를 두 번 돌린다.
• 그런데, 난 더 최적화하기 위해서 S->L, L->E로 돌리지 않고, S->L, E->L로 돌렸다. 이렇게 하면 출구가 X로 봉쇄되어 있는 경우 더 빨리 탈출할 수 있기 때문이다.

코드

from collections import deque

#상하좌우
dy = [-1, 1, 0, 0]
dx = [0, 0, -1, 1]

def solution(maps):
    answer = 0
    row = len(maps)
    col = len(maps[0])
    #일단 시작점 좌표를 찾자
    start = []
    end = []
    for i in range(row):
        for j in range(col):
            if maps[i][j] == 'S':
                start = [i, j]
            elif maps[i][j] == 'E':
                end = [i, j]
            
            if start and end:
                break
                
    def bfs(target):
        nonlocal start
        visited = [[False] * col for _ in range(row)]
        queue = deque()
        queue.append([start[0], start[1], 0])
        visited[start[0]][start[1]] = True
        while queue:
            r, c, level = queue.popleft()
            if maps[r][c] == target:
                start = end
                return level
                break
            #상하좌우 탐색 후 이동
            for i in range(4):
                nr, nc = r + dy[i], c + dx[i]
                if nr >= 0 and nc >= 0 and nr < row and nc < col and not visited[nr][nc] and maps[nr][nc] != 'X':
                    queue.append([nr, nc, level+1])
                    visited[nr][nc] = True
        
        return 0
    
    lever_cost = bfs('L')
    if lever_cost == 0:
        return -1
    else:
        answer += lever_cost
    
    exit_cost = bfs('L')
    if exit_cost == 0:
        return -1
    else:
        answer += exit_cost
    
    return answer