[프로그래머스 Lv2] 파이썬 - 미로 탈출

프로그래머스 - 미로 탈출

# 조건

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

제한사항

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

# 접근 방법

• 전형적인 bfs 문제이다.
• start 에서 레버를 찾을 때 까지 이동한 거리 + 레버에서 출구까지의 이동한 거리를 더해주면 된다.
• 따라서 레버에 도착하였을 때 visited 배열을 다시 초기화 해준 후 bfs를 한번 더 돌리면 되는 것이다.
• bfs를 재사용하기 위하여, 현재 레버를 찾는지 출구를 찾는지 0과 1로 표시를 해준다.

from collections import deque

# num = 0 이면 레버 찾기, 1이면 출구 찾기
def bfs(st, arr, num):
    di, dj = [1, -1, 0, 0], [0, 0, 1, -1]
    visited = [[-1] * M for _ in range(N)]
    visited[st[0]][st[1]] = 0
    q = deque()
    q.append((st[0], st[1]))
    while q:
        si, sj = q.popleft()
        if num == 0:
            if si == lever[0] and sj == lever[1]:
                return visited[si][sj]
        elif num == 1:
            if si == end[0] and sj == end[1]:
                return visited[si][sj]

        for d in range(4):
            ni, nj = si + di[d], sj + dj[d]
            if 0<=ni<N and 0<= nj<M and visited[ni][nj] == -1 and arr[ni][nj] != 'X':
                q.append((ni, nj))
                visited[ni][nj] = visited[si][sj] + 1
    return 0

def solution(maps):
    global N, M, start, end, lever
    N = len(maps)
    M = len(maps[0])
    answer = 0
    # 레버를 당길 수 있는지 여부
    for i in range(N):
        for j in range(M):
            if maps[i][j] == 'S':
                start = [i, j]
            elif maps[i][j] == 'E':
                end = [i, j]
            elif maps[i][j] == 'L':
                lever = [i, j]

    # 레버 당기기
    for k in range(2):
        if k == 0:
            flag = bfs(start, maps, k)
        else:
            flag = bfs(lever, maps, k)
        if not flag:
            return -1
        answer += flag
    return answer