[파이썬] 2178 - 미로 탐색

문제

N×M크기의 배열로 표현되는 미로가 있다.

1	0	1	1	1	1
1	0	1	0	1	0
1	0	1	0	1	1
1	1	1	0	1	1

미로에서 1은 이동할 수 있는 칸을 나타내고, 0은 이동할 수 없는 칸을 나타낸다. 이러한 미로가 주어졌을 때, (1, 1)에서 출발하여 (N, M)의 위치로 이동할 때 지나야 하는 최소의 칸 수를 구하는 프로그램을 작성하시오. 한 칸에서 다른 칸으로 이동할 때, 서로 인접한 칸으로만 이동할 수 있다.

위의 예에서는 15칸을 지나야 (N, M)의 위치로 이동할 수 있다. 칸을 셀 때에는 시작 위치와 도착 위치도 포함한다.

 

풀이

 먼저, 문제에 있는 표를 배열로 나타내어 그래프라고 생각하고, 시작 노드인 (0, 0)부터 상, 하, 좌, 우로 탐색을 시작한다.

 알고리즘 포스팅에서 연결되어있는 노드를 탐색하기 위해 range를 사용하여 1번노드부터 차례대로 탐색했는데, 이번엔 시작노드 matrix[a][b]를 기준으로 matrix[a+1][b], matrix[a-1][b], matrix[a][b+1], matrix[a][b-1]을 탐색한다.

 방문한 노드도 matrix와 동일한 크기의 배열로 만들어 방문한 위치를 몇번째로 방문했는지 표시해준다.

 

from collections import deque
#초기화
N, M = map(int,input().split())
matrix = [[0]*M for i in range(N)]
visit = [[0]*M for i in range(N)]
for i in range(N):
    a = input().strip()
    for b, c in enumerate(a):
        matrix[i][b] = int(c)
dx = [-1, 1, 0, 0]
dy = [0, 0, -1, 1]
queue = deque()

#BFS 함수
def BFS(y, x, visited):
    count = 1
    visited[y][x] = count
    queue.append([y,x,count])
    while queue:
        y, x, count = queue.popleft()
        count += 1
        for i in range(4):
            nx = x + dx[i]
            ny = y + dy[i]
            if 0 <= nx < M and 0 <= ny < N:
                if matrix[ny][nx] and not visited[ny][nx]:
                    visited[ny][nx] = count
                    queue.append([ny,nx,count])
    return visited[N-1][M-1]
    
print(BFS(0, 0, visit))

평가

BFS탐색(미로찾기)의 가장 기본이 되는 문제다.