백준 2178 - 미로 탐색 (Python3)
코드
# 최단 경로 => BFS
from collections import deque
n, m = map(int, input().split())
graph = []
for _ in range(n):
graph.append(list(map(int, input())))
# 이동 방향을 의미하는 리스트
dx = [-1, 1, 0, 0]
dy = [0, 0, -1, 1]
def bfs(x, y):
q = deque()
q.append((x, y))
while q:
x, y = q.popleft()
# 4방향으로 이동하므로 for문 사용
for i in range(4):
nx = x + dx[i]
ny = y + dy[i]
# 미로를 벗어나면 반복 스킵
if nx < 0 or nx >= n or ny < 0 or ny >= m:
continue
# 1로만 지나갈 수 있으므로 0 제외
if graph[nx][ny] == 0:
continue
# 좌표값이 1이므로 그냥 편하게
# 이전 좌표에 + 1을 해주는 로직으로
# 몇칸을 이동했는가 계산
if graph[nx][ny] == 1:
graph[nx][ny] = graph[x][y] + 1
q.append((nx, ny))
# BFS 실행 => 0,0이 시작
bfs(0, 0)
# 맨 마지막 칸의 좌표
print(graph[n-1][m-1])
문제 해설
대표적인 BFS 문제이다.
최단 경로를 구하라는 문제는 DFS가 아닌 BFS로 풀어야 확실하게 해결된다.
대부분의 BFS문제는 기본 로직이 비슷하므로 디테일한 설명은 생략하고, 주석으로 대신하겠습니다!!
댓글남기기