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

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

프로그래머스

 

Status : Solved

 

Time : 00:19:26

 

---

 

문제 설명

 

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

 

 

---

 

풀이

 

경로 탐색 문제 기본판. 우선 출구는 레버를 당겨야 열리므로 입구 -> 레버, 레버-> 출구 순으로 총 2회 경로 탐색을 시행한다. 만약 둘 중 하나라도 불가능하다면 -1을 출력하도록 하자.

 

• find_points : maps 전체를 탐색하며 시작 좌표와 레버 좌표를 반환한다.
• bfs : 경로 탐색 함수. 특정 시작점에서 맵상의 특정 타겟까지 도달할 때까지 탐색을 수행한다. 만약 도달할 수 없다면 -1을 반환한다.
• solution : 메인함수. 두 번의 경로탐색을 시행하고, 둘 중 하나라도 -1이라면 -1을 반환, 그렇지 않다면 그 합을 반환한다.

 

 

풀이 코드

 

from collections import deque

dx = [-1, 1, 0, 0]
dy = [0, 0, -1, 1]

def find_points(maps):
    col, row = len(maps), len(maps[0])
    
    for i in range(col) :
        for j in range(row) :
            if maps[i][j] == 'S' :
                start = (j, i)
            if maps[i][j] == 'L' :
                lever = (j, i)
    
    return start, lever

def dfs(start, maps, target) :
    col, row = len(maps), len(maps[0])
    x, y = start
    visited = [[False]*row for _ in range(col)]
    visited[y][x] = True
    q = deque([(x, y, 0)])

    while q :
        x, y, dist = q.popleft()
        if maps[y][x] == target :
            return dist
        
        for k in range(4) :
            ax, ay = x + dx[k], y + dy[k]
            if -1 < ax < row and -1 < ay < col and not visited[ay][ax] and maps[ay][ax] != 'X' :
                visited[ay][ax] = True
                q.append((ax, ay, dist+1))
        
    return -1

def solution(maps):    
    start, lever = find_points(maps)
    lever_dist = dfs(start, maps, 'L')
    end_dist = dfs(lever, maps, 'E')

    return lever_dist + end_dist if lever_dist > -1 and end_dist > -1 else -1

풀이 완료!!