[백준 2178번] 미로 탐색 (python)

문제

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

 

기록

(graph 2일차) 오늘 진짜 메모리초과 시간초과에 엄청 시간들였다. 해결방법은 큐에 불필요한 노드들이 중복으로 들어가지 않게 하는 것이다. 이를 위해서는 인접노드를 큐에 추가할 때, 방문했다고 표시도 함께 해야했다.

 

이번 문제는 최단 거리를 구해야 하는 문제이기 때문에 bfs로 해결했다. 출발지부터 한단계 이동할 때마다 1씩 큰 수를 이차원 배열에 저장해나가며 목적지까지의 가장 작은 수를 구하는 방법이다.

 

코드

import sys
from collections import deque
input = sys.stdin.readline

n, m = map(int, input().split())
board = [list(map(int, input().strip())) for i in range(n)]
queue = deque([(0, 0)])
visit = [[0]*m for i in range(n)]
count = [[0]*m for i in range(n)]
visit[0][0] = 1
count[0][0] = 1

while queue:
    y, x = queue.popleft()
    for i, j in [(1, 0), (0, 1), (-1, 0), (0, -1)]:
        if y+i>=0 and y+i<n and x+j>=0 and x+j<m and visit[y+i][x+j]==0 and board[y+i][x+j]:
            queue.append((y+i, x+j))
            count[y+i][x+j] = count[y][x]+1
            visit[y+i][x+j] = 1
        if y+i==n-1 and x+j==m-1 : break
print(count[n-1][m-1])

1. 변수와 보드 저장, 큐와 방문여부를 저장할 배열, 출발지부터의 거리를 저장할 배열을 초기화한다.

2. 큐에서 왼쪽노드를 뽑아 각 인덱스의 범위와 방문여부, 이동가능여부를 확인하여 큐에 추가하고 방문표시, 거리를 저장한다. (시간 줄이기 위해서 추가로 인접노드가 도착점일 때 멈추도록 했는데 빼도 상관없다.)

3. 출발지부터 도착지까지 최소거리를 저장한 배열에서 마지막 칸을 출력한다.