[이것이취업을위한코딩테스트다] chap05 - 미로 탈출

미로 탈출

 

N X M크기의 직사각형 형태의 미로가 있다. 최초 위치는 (1,1)이고 미로의 출구는 (N,M)이며 한 번에 한 칸씩 이동할 수 있다. 괴물이 있는 부분은 0, 없는 부분은 1로 괴물을 피해 탈출하기 위해 움직여야 하는 최소 개수를 구하시오. 칸을 셀 때는 시작 칸과 마지막 칸을 모두 포함해서 계산한다.

 

코드

# 미로 탈출

from collections import deque

n, m = map(int, input().split())

graph = []
for i in range(n):
    graph.append(list(map(int, input())))

dx = [-1, 1, 0, 0]
dy = [0, 0, -1, 1]

def bfs(x, y):
    queue = deque()
    queue.append((x, y))

    while queue:
        x, y = queue.popleft()

        for i in range(4):
            nx = x + dx[i]
            ny = y + dy[i]
            
            if nx < 0 or ny < 0 or nx >= n or ny >= m:
                continue

            if graph[nx][ny] == 0:
                continue

            if graph[nx][ny] == 1:
                graph[nx][ny] = graph[x][y]+1
                queue.append((nx, ny))

    return graph[n-1][m-1]

print(bfs(0,0))

 

문제풀이

 

1) bfs를 활용해 상,하,좌,우를 살피며 1이 있는 부분으로 탐색을 진행한다.

2) 1이 있는 부분에 도착했을 때 해당 부분에 +1을 진행한다.

3) 마지막 칸 바로 전에 도착했을 때의 값을 출력한다.