[백준 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)의 위치로 이동할 수 있다. 칸을 셀 때에는 시작 위치와 도착 위치도 포함한다.

입력

첫째 줄에 두 정수 N, M(2 ≤ N, M ≤ 100)이 주어진다. 다음 N개의 줄에는 M개의 정수로 미로가 주어진다. 각각의 수들은 붙어서 입력으로 주어진다.

출력

첫째 줄에 지나야 하는 최소의 칸 수를 출력한다. 항상 도착위치로 이동할 수 있는 경우만 입력으로 주어진다.

풀이

최소한의 칸을 이동하여 마지막 칸까지 이동해야 한다.

입력받은 칸에 대한 정보와 방문한 칸을 표시하기 위해 matrix와 visit, 두개의 리스트를 생성한다.

칸의 움직임은 위, 아래, 왼쪽, 오른쪽 으로 나타낼 수 있기 때문에

dx = [-1,1,0,0], dy = [0,0,-1,1] 로 생성한다.

이제 bfs를 이용하여 문제를 해결하면 되는데 bfs는 queue를 통해 구현한다.

먼저 초기값인 0,0을 큐에 넣는다.

이후 큐가 남아있을 때까지 큐를 pop한 값의 인접노드를 검사하면 된다.

dx, dy로 상하좌우의 인접노드를 검사하고 인접노드에 갈 수 있을 때,

visit에는 지금까지 지나온 칸의 수를 더해준다.

x가 n-1, y가 m-1 즉 마지막 칸에 도착했을 때 이동거리를 출력하고 반복문을 종료한다.

 

import sys
n, m = map(int, sys.stdin.readline().rstrip("\n").split())
matrix = []
for i in range(n):
    temp = list(map(int, sys.stdin.readline().rstrip("\n")))
    matrix.append(temp)
visit=[[0]*m for _ in range(n)]

dx = [-1,1,0,0]
dy = [0,0,-1,1]
queue=[(0,0)]
visit[0][0]=1

while queue:
    x,y=queue.pop(0)
    if x==n-1 and y==m-1:
        print(visit[x][y])
        break
    for i in range(4):
        x_ = x + dx[i]
        y_ = y + dy[i]
        if(x_>=0 and x_<n and y_>=0 and y_<m):
            if(visit[x_][y_] == 0 and matrix[x_][y_]==1):
                visit[x_][y_] = visit[x][y] + 1
                queue.append((x_,y_))