프로그래머스 [level2] 미로 탈출 - python3

미로 탈출

문제 설명

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

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

제한사항

• 5 ≤ maps의 길이 ≤ 100
  • 5 ≤ maps[i]의 길이 ≤ 100
  • maps[i]는 다음 5개의 문자들로만 이루어져 있습니다.
    • S : 시작 지점
    • E : 출구
    • L : 레버
    • O : 통로
    • X : 벽
  • 시작 지점과 출구, 레버는 항상 다른 곳에 존재하며 한 개씩만 존재합니다.
  • 출구는 레버가 당겨지지 않아도 지나갈 수 있으며, 모든 통로, 출구, 레버, 시작점은 여러 번 지나갈 수 있습니다.

입출력 예

maps	result
["SOOOL","XXXXO","OOOOO","OXXXX","OOOOE"]	16
["LOOXS","OOOOX","OOOOO","OOOOO","EOOOO"]	-1

입출력 예 설명

입출력 예 #1

주어진 문자열은 다음과 같은 미로이며

다음과 같이 이동하면 가장 빠른 시간에 탈출할 수 있습니다.

4번 이동하여 레버를 당기고 출구까지 이동하면 총 16초의 시간이 걸립니다. 따라서 16을 반환합니다.

입출력 예 #2

주어진 문자열은 다음과 같은 미로입니다.

시작 지점에서 이동할 수 있는 공간이 없어서 탈출할 수 없습니다. 따라서 -1을 반환합니다.

문제 풀이

from collections import deque

def solution(maps):
    def find_L(y, x):
        que = deque()
        que.append((y, x, 0))
        while que:
            y, x, cnt = que.popleft()
            if maps[y][x] == "L":
                yxc = [y, x, cnt]
                return yxc
            for i in range(4):
                xx = x + dx[i]
                yy = y + dy[i]
                if 0 <= xx < len(maps[0]) and 0 <= yy < len(maps):
                    if visited_map[yy][xx] == 0 and maps[yy][xx] != "X":
                        visited_map[yy][xx] = 1
                        que.append((yy, xx, cnt + 1))

    def find_E(y, x):
        que = deque()
        que.append((y, x, 0))
        while que:
            y, x, cnt = que.popleft()
            if maps[y][x] == "E":
                return cnt
            for i in range(4):
                xx = x + dx[i]
                yy = y + dy[i]
                if 0 <= xx < len(maps[0]) and 0 <= yy < len(maps):
                    if visited_map[yy][xx] == 0 and maps[yy][xx] != "X":
                        visited_map[yy][xx] = 1
                        que.append((yy, xx, cnt + 1))

    answer = 0
    dx = [-1, 1, 0, 0]
    dy = [0, 0, -1, 1]
    visited_map = [[0 for _ in range(len(maps[0]))] for _ in range(len(maps))]
    lever_map = [[0 for _ in range(len(maps[0]))] for _ in range(len(maps))]
    for i in range(len(maps)):
        for j in range(len(maps[0])):
            if maps[i][j] == "S":
                a = find_L(i, j)
                # print(len(a))
                if a is None:
                    answer = -1
                    return answer
                else:
                    answer = a[2]
                    # print(a)
                    # # print(answer)
                    visited_map = [[0 for _ in range(len(maps[0]))] for _ in range(len(maps))]
                    b = find_E(a[0], a[1])
                    if b is None:
                        answer = -1
                        return answer
                    else:
                        # print(b)
                        answer = answer + b

    return answer

문제 해석

BFS를 2번 사용해서 처음에 레버를 찾고 이후 방문 노드를 초기화 한 다음, 다시 탈출구를 찾는 방식으로 탐색을 진행했다. 이후 탐색했을 때 횟수를 각자 더해서 진행했다.