[프로그래머스] 미로 탈출 (파이썬)

문제

코딩테스트 연습 - 미로탈출

문제 설명

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

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

제한사항

🧭 접근

탐색류 문제이므로, BFS를 사용하면 되겠다고 생각했다.

✔ 풀이

1. maps의 x,y 크기 및 필요한 인덱스 구하기
2. BFS 함수 작성
3. BFS를 이용하여 시작 -> 레버, 레버 -> 출구 까지의 거리 구하기

📙 소스 코드

from collections import deque

def bfs(maps,y,x,dy,dx):
    q = deque()
    q.append([y,x])
    cnt = 1
    time = 0
    movex = [0,0,-1,1] # 상 하 좌 우 
    movey = [-1,1,0,0]
    dis = [[0] * maxx for _ in range(maxy)] # 거리 배열
    visited = [[0] * maxx for _ in range(maxy)] 
    visited[y][x] = 1
    b_flag = 0
    while q:
        if b_flag == 1:
            break
        ny,nx = q.popleft()
        for i in range(4): # 4방향 이동
            mx = nx+movex[i]; my = ny+movey[i]
            # 인덱스 벗어날경우
            if mx < 0 or my < 0 or mx >= maxx or my >= maxy: 
                continue
            # 벽이나 이미 방문한 곳 일경우
            if maps[my][mx] == 'X' or visited[my][mx] == 1:
                continue
            q.append([my,mx])
            visited[my][mx] = 1
            dis[my][mx] = dis[ny][nx] + 1 # 시작점부터의 거리 계산
            if my == dy and mx == dx: # 목표 도달한경우 break
                b_flag = 1
                break


    if b_flag != 1: # 목표 도달 못한경우
        return -1
    return dis[dy][dx] # 목표까지의 거리 반환

def solution(maps):
   # S : 시작 지점 E : 출구 L : 레버 O : 통로 X : 벽
    global maxx
    global maxy
    maxx = len(maps[0]) # x 최고 인덱스
    maxy = len(maps) # y 최고 인덱스

    for i in range(maxy): # 시작점, 출구, 레버 인덱스 구하기
        for j in range(maxx):
            val = maps[i][j]
            if val == 'S':
                start = [i,j]
            elif val == 'E':
                exit = [i,j]
            elif val == 'L':
                lever = [i,j]
	# 시작 -> 레버
    res = bfs(maps,start[0],start[1],lever[0],lever[1]) 
    if res == -1:
        return -1
    # 레버 -> 출구
    res2 = bfs(maps,lever[0],lever[1],exit[0],exit[1])
    if res2 == -1:
        return -1

    return res + res2 # 거리 반환

🤔 느낀 점

탐색 문제류는 많이 풀어봐야 알고리즘을 빠르게 찾아 적용시킬 수 있고, 코드 작성 시간도 빨라지는 것 같다.
예전보다는 많이 빨라졌음 😁