[프로그래머스] (Lv.2) 미로 탈출 [Python]

문제 링크

프로그래머스

문제

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

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

코드

from collections import deque

def solution(maps):
    answer = float("inf")
    n = len(maps)
    m = len(maps[0])
    sx, sy = 0, 0
    for i in range(n):
        for j in range(m):
            if maps[i][j] == "S":
                sx, sy = i, j
                break
    
    visited = set()
    visited.add((sx, sy))
    queue = deque()
    queue.append((sx, sy, 0))
    dx = [1, 0, -1, 0]
    dy = [0, 1, 0, -1]
    flag = False
    while queue:
        x, y, time = queue.popleft()
        if maps[x][y] == 'L':
            flag = True
            break
        for i in range(4):
            nx = x + dx[i]
            ny = y + dy[i]
            if 0 <= nx <= n-1 and 0 <= ny <= m-1:
                if maps[nx][ny] != 'X' and (nx, ny) not in visited:
                    visited.add((nx, ny))
                    queue.append((nx, ny, time+1))
    
    if not flag:
        return -1
    
    visited = set()
    visited.add((x, y))
    queue.clear()
    queue.append((x, y, time))
    while queue:
        x, y, time = queue.popleft()
        if maps[x][y] == 'E':
            answer = time
            break
        visited.add((x, y))
        for i in range(4):
            nx = x + dx[i]
            ny = y + dy[i]
            if 0 <= nx <= n-1 and 0 <= ny <= m-1:
                if maps[nx][ny] != 'X' and (nx, ny) not in visited:
                    visited.add((nx, ny))
                    queue.append((nx, ny, time+1))
        
    return answer if answer < float("inf") else -1

• 모든 이동 가능한 칸은 여러 번 방문할 수 있기 때문에, 'L'까지의 최단경로를 찾는 BFS, 'E'까지의 최단경로를 찾는 BFS, 총 2번의 탐색을 시행해야 했다.
• visited를 2차원 배열로 선언하면 시간 초과가 발생하기 때문에 해시 기반인 set을 사용하여 visited 처리를 if문을 만족할 때만 수행하게 하여 제한 시간을 만족시킬 수 있었다.