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

문제 설명

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

---

풀이

이런 미로탈출은 dfs 대표문제인듯. '큐에 넣고 빼면서 해당 좌표에서 갈 수 있는 좌표 찾아 넣기' 반복

시작, 레버, 끝점 좌표를 찾아내고 시작->레버, 레버->끝점을 각각 dfs 돌렸다. 한번에 하면 시간초과됨 ㅠ

근데 오랜만에 구현하니 버벅거렸다. 코드가 깔끔하지는 않아서 다시 정리해야겠다.

 

from collections import deque

def dfs(startX, startY, endX, endY, maps):
    visited = [[0 for _ in range(len(maps[0]))] for _ in range(len(maps))]
    rows, cols = len(maps), len(maps[0])
    dx = [-1, 1, 0, 0]
    dy = [0, 0, -1, 1]
    queue = deque()
    queue.append((startX, startY))

    while queue:
        x, y = queue.popleft()

        if (x, y) == (endX, endY):
            return visited[x][y]

        for i in range(4):
            nx = x + dx[i]
            ny = y + dy[i]

            if nx < 0 or ny < 0 or nx >= rows or ny >= cols:
                continue
            if maps[nx][ny] == 'X':
                continue
            if visited[nx][ny] == 0:
                visited[nx][ny] = visited[x][y] + 1
                queue.append((nx, ny))

    if not queue:
        return -1
    x, y = queue.pop()

    return visited[x][y]

def solution(maps):
    x, y, l_x, l_y, e_x, e_y = -1, -1, -1, -1, -1, -1

    for i, m in enumerate(maps):
        for j, n in enumerate(m):
            if n == 'S':
                x, y = i, j
            if n == 'E':
                e_x, e_y = i, j
            if n == 'L':
                l_x, l_y = i, j
        if x > -1 and l_x > -1 and e_x > -1:
            break

    r1 = dfs(x, y, l_x, l_y, maps)
    r2 = dfs(l_x, l_y, e_x, e_y, maps)
    if r1 == -1 or r2 == -1:
        return -1
    else:
        return r1 + r2