[Python] [백준🥈1] 2178번 미로 탐색 (BFS 알고리즘)

https://www.acmicpc.net/problem/2178

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개의 정수로 미로가 주어진다. 각각의 수들은 붙어서 입력으로 주어진다.

 

출력

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

 

풀이

import sys
from collections import deque # 큐 구현을 위해 데크 사용

def bfs(graph, visited):
    queue = deque([[0, 0]]) # 2차원 리스트의 인덱스를 고려하여 (0, 0)에서 시작
    visited[0][0] = True    # 큐에 삽입할 때 방문처리
    while graph[n-1][m-1] == 1: # (n, m)에 처음으로 도착할 때, 즉 이동이 최소일 때 종료
        y, x = queue.popleft() # 행을 y, 열을 x로 설정
        dy = [y-1, y+1, y, y] # (y, x)의 4방향
        dx = [x, x, x-1, x+1]
        for i in range(4):
            if dy[i] > -1 and dy[i] < n and dx[i] > -1 and dx[i] < m and graph[dy[i]][dx[i]] != 0 and not visited[dy[i]][dx[i]]:
                queue.append([dy[i], dx[i]]) # 그래프를 벗어나지 않고 이동 불가능(0)한 블럭 제외 그리고 재방문 하지 않은 블럭
                visited[dy[i]][dx[i]] = True
                graph[dy[i]][dx[i]] = graph[y][x] + 1 # 이동할 블럭에 이전 이동한 블럭 수 + 1

input = sys.stdin.readline
n, m = map(int, input().split())
graph = [list(map(int, input().rstrip())) for i in range(n)]
visited = [[False] * m for i in range(n)]
bfs(graph, visited)
print(graph[n-1][m-1])

 

다른 풀이 참조 및 개선할 점

코드를 작성하다 보니 몇몇 경우에는 블럭을 빙글빙글 도는 무한루프에 빠져서 BFS 함수의 매개변수로 visited 리스트를 선언하고 입력 받게 추가하였다. 이방법 대신에 이전에 방문한 블럭들을 0으로 설정하여 재방문을 방지하고 코드를 더 간결하게 작성할 수 있겠다.