본문 바로가기

알고리즘

BFS) 미로 탈출

출발위치 : (0,0)

도착위치: (n-1,m-1) 미로의 맨 오른쪽 아래

벽 : 1

길: 0

from collections import deque

# N, M을 공백으로 입력받기
n,m = map(int,input().split())

# 2차원 리스트의 맵 정보 입력받기
graph = []
for i in range(n):
  graph.append(list(map(int, input())))

# 이동한 네 방향 정의 (상, 하, 좌, 우)
dx = [-1, 1, 0, 0]
dy = [0, 0, -1, 1]

# BFS 소스코드 구현
def bfs(x, y):
  # 큐 구현을 위해 deque 라이브러리 사용
  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 nx >= n or ny < 0 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일 경우) 최단 거리 기록(이전 노드 + 1)

'알고리즘' 카테고리의 다른 글

이진 탐색 - 구현  (0) 2023.09.25
퀵 정렬  (0) 2023.09.22
DFS) 음료수 얼려 먹기  (0) 2023.09.22
BFS  (0) 2023.09.22
DFS  (0) 2023.09.22