백준 14940 [쉬운 최단거리] 파이썬

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

오랜만에 BFS문제를 마주쳤더니 기억이 잘 나질 않는군요..

 

#백준 14940 쉬운 최단거리
import sys
dr=[-1,0,1,0]
dc=[0,1,0,-1]
n,m= map(int,sys.stdin.readline().rstrip().split())
map_matrix= []
visited_matrix=[[-1 for _ in range(m)] for _ in range(n)]
q=[]
unvisitable_points=[]
start_r=0
start_c=0

for i in range(n):
    row=list(map(int,sys.stdin.readline().rstrip().split()))
    for j in range(m):
        if row[j]==2:
            start_r=i
            start_c=j
    map_matrix.append(row)
q.append((start_r,start_c))
visited_matrix[start_r][start_c]=0

# BFS START
while q:
    r,c=q.pop(0)
    for i in range(4):
        R=r+dr[i]
        C=c+dc[i]
        if 0<=R<n and 0<=C<m and visited_matrix[R][C]==-1:
            if map_matrix[R][C]!=0:
                visited_matrix[R][C]=visited_matrix[r][c]+1
                q.append((R,C))
#BFS END

for r in range(n):
    for c in range(m):
        if map_matrix[r][c]==0:
            visited_matrix[r][c]=0

for i in range(n):
    print(*visited_matrix[i])

BFS 잘 구현하고,

전체 맵에서 0인 부분을 0으로 표기 (중요) 하면 해결.