[Python/Programmers] 미로 탈출

https://school.programmers.co.kr/learn/courses/30/lessons/159993#

프로그래머스

문제 설명

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

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

풀이

Code

from collections import deque
def solution(maps):
    # 1. bfs 함수 정의
    def bfs(x, y, destination) :
        # 1-1. 큐에 (현재 위치, 거리) 삽입
        queue = deque()
        queue.append((x, y, 0))
        # 1-2. 방문 여부 리스트 생성
        visited = [[False] * m for _ in range(n)]
        # 1-3. 현재 위치 방문 처리
        visited[x][y] = True
        # 1-4.
        while queue :
            # 1-4-1. 위치, 거리 반환
            x, y, d = queue.popleft()
            # 1-4-2.
            for dir_x, dir_y in [(-1, 0), (1, 0), (0, -1), (0, 1)] :
                # 다음 위치 설정
                nx, ny = x + dir_x, y + dir_y
                # 예외 처리
                if nx < 0 or nx >= n or ny < 0 or ny >= m : continue
                # 목표 지점일 경우
                if [nx, ny] == destination : return d + 1
                # 방문하지 않은 경우
                if not visited[nx][ny] and maps[nx][ny] != 'X':
                    # 방문 처리
                    visited[nx][ny] = True
                    # 큐에 (다음 위치, 거리 + 1) 삽입
                    queue.append((nx, ny, d + 1))
        # 1-5. -1 반환
        return -1
    n, m = len(maps), len(maps[0])
    # 2. 시작 지점, 레버 위치, 탈출 지점 인덱스 구하기
    for i in range(n) :
        for j in range(m) :
            if maps[i][j] == 'S' : start = [i, j]
            elif maps[i][j] == 'L' : lever = [i, j]
            elif maps[i][j] == 'E' : end = [i, j]
    # 3. [시작지점-레버 위치 간 거리, 레버 위치-탈출 지점 간 거리] 구하기
    distance = [bfs(start[0], start[1], lever), bfs(lever[0], lever[1], end)]
    # 4. 결과 출력
    return -1 if -1 in distance else sum(distance)