이코테 2021, BFS/미로 탈출 (파이썬)

문제 정의

동빈이는 N x M 크기의 직사각형 형태의 미로에 갇혔습니다. 미로에는 여러 마리의 괴물이 있어 이를 피해 탈출해야합니다.

 

동빈이의 위치는 (1, 1)이며 미로의 출구는 (N, M)의 위치에 존재하며 한 번에 한 칸씩 이동할 수 있습니다. 이때 괴물이 있는 부분은 0으로, 괴물이 없는 부분은 1로 표시되어있습니다. 미로는 반드시 탈출할 수 있는 형태로 제시됩니다.

 

이때 동빈이가 탈출하기 위해 움직여야하는 최소 칸의 개수를 구하세요. 칸을 셀 때는 시작 칸과 마지막 칸을 모두 포함해서 계산합니다

 

 

입력

첫 번째 줄에 두 정수 N, M이 주어진다. (4 <= N, M <= 200) 다음 N개의 줄에는 각각 M개의 정수(0 혹은 1)로 미로의 정보가 주어진다. 각각의 수는 공백 없이 붙어서 입력으로 제시된다. 또한 시작 칸과 마지막 칸은 항상 1이다.

 

 

출력

첫째 줄에 최소 이동 칸의 개수를 출력한다.

 

 

예제 입력 1

5 6
101010
111111
000001
111111
111111

 

예제 출력 1

10

 

 

 

접근 방법

1. 큐에 현재 위치에서 동서남북에 해당하는 위치를 다 넣고 빼는 과정을 반복하며 count를 하나씩 증가시킨다.
2. n-1, m-1 위치에 도달할 경우 count를 리턴한다.

 

 

코드

from collections import deque
n, m = 5, 6
map_ = [[1, 0, 1, 0, 1, 0], [1, 1, 1, 1, 1, 1], [0, 0, 0, 0, 0, 1], [1, 1, 1, 1, 1, 1], [1, 1, 1, 1, 1, 1]]


# 1. 큐에 현재 위치에서 동서남북에 해당하는 위치를 다 넣고 빼는 과정을 반복하며 count를 하나씩 증가시킨다.
# 2. n-1, m-1 위치에 도달할 경우 count를 리턴한다.


def bfs(n, m):
    queue = deque()
    depth = 0
    queue.append([0,0,depth])
    map_[0][0] = 0
    while queue:
        col, row, depth = queue.popleft()
        depth += 1
        for dcol, drow in [[0, 1], [0, -1], [1, 0], [-1, 0]]:
            ncol, nrow = dcol+col, drow+row
            if 0 <= ncol <= n-1 and 0 <= nrow <= m-1 and map_[ncol][nrow] == 1:
                queue.append([ncol, nrow, depth])
                map_[ncol][nrow] = 0
                

        if col == n-1 and row == m-1:
            return depth

print(bfs(n,m))

- 초창기에 해결한 BFS 문제이기에 다소 코드가 난잡하다.