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

나의 접근

1. 인접한 칸 이동 -> 동서남북 이동가능
2. 최소의 이동 -> 탐색 시 가장 먼저 발견했을 때의 이동 횟수

=> BFS로 탐색하여 1,1에서 각 칸까지의 최소 거리를 저장

코드

from collections import deque
N, M = map(int, input().split())

def bfs(maze, visited, distance, x, y):

    def setValue(visited, distance, a, b ,da, db):
        visited[a+da][b+db] = True
        distance[a+da][b+db] = distance[a][b] + 1
        return

    queue = deque([[x, y]])
    distance[x][y] = 1
    visited[x][y] = True

    while queue:
        a, b = queue.popleft()
        if(a-1 >= 0 and not visited[a-1][b] and maze[a-1][b] == 1):

            setValue(visited,distance,a,b,-1,0)
            queue.append([a-1, b])
        if(a+1 < N and not visited[a+1][b] and maze[a+1][b] == 1):
            setValue(visited, distance,a,b,+1,0)
            queue.append([a+1,b])

        if(b-1>=0 and not visited[a][b-1] and maze[a][b-1] == 1):
            setValue(visited, distance, a, b, 0, -1)
            queue.append([a , b-1])

        if(b+1<M and not visited[a][b+1] and maze[a][b+1] == 1):
            setValue(visited,distance, a, b, 0, 1)
            queue.append([a, b+1])
maze = []
for i in range(N):
    maze.append(list(map(lambda x: int(x),input())))

visited = [ [False]*M for _ in range(N) ]
distance = [[0] *M for _ in range(N)]

bfs(maze, visited, distance, 0, 0)
print(distance[N-1][M-1])

소요 시간

25분