백준 온라인 저널, BFS/2178번 : 미로 탐색(파이썬)

2021. 6. 15. 02:44알고리즘/BFS

728x90
반응형

문제 정의

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

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

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

 

 

입력

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

 

 

출력

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

 

 

예제 입력 1

4 6
101111
101010
101011
111011

예제 출력 1

15

 

예제 입력 2

4 6
110110
110110
111111
111101

예제 출력 2

9

- BFS의 기본 문제

 

 

접근 방법

1. 시작점을 기준으로 위치(인덱스 기준)를 큐에 삽입한다.

2. 큐의 값을 pop하고 해당 위치를 기준으로 동서남북으로 움직일 경우 이동 가능한 칸(값이 1인 경우)을 찾아 큐에 삽입하고 해당 위치의 값을 이전 위치의 값 + 1로 바꾼다.

 

 

코드

from collections import deque
import sys

n, m = map(int, sys.stdin.readline().split())
map_ = [list(map(int, list(str(sys.stdin.readline()))[:m])) for _ in range(n)]
step_ = [[0, 1], [0, -1], [1, 0], [-1, 0]] # 동서남북

def bfs():
    queue = deque()
    queue.append([0, 0])
    
    while queue:
        x, y = queue.popleft()
        
        for dx, dy in step_:
            if 0 <= x+dx <= n-1 and 0 <= y+dy <= m-1 and map_[x+dx][y+dy] == 1:
                queue.append([x+dx, y+dy])
                map_[x+dx][y+dy] = map_[x][y] + 1

bfs()
print(map_[n-1][m-1])

 

 

728x90
반응형