[DFS/BFS] DFS/BFS (3) 실전 문제 - 미로 탈출 ⭐

이코테 p.152 실전문제 - 미로 탈출 ((0, 0)부터 (n - 1, m - 1)까지의 최단 경로)

최단거리 구하기 ➡️ BFS

경우의수 구하기 ➡️ DFS

입력 예시

5 6 101010 111111 000001 111111 111111

출력 예시

10

교재 소스코드

• (0, 0)에서 (n - 1, m - 1)까지의 최단 거리니까 BFS 사용
• (0, 0)에서 해당 노드까지의 거리를 graph[x][y]에 저장
  • 현재 노드의 값을 (바로 전 노드의 값 + 1)로 계속해서 업데이트
• 최종적으로 graph[n - 1][m - 1]에는 (0, 0)에서부터의 최단거리가 저장됨

from collections import deque

# 입력 받기
n, m = map(int, input().split())
graph = []
for _ in range(n):
  graph.append(list(map(int, input())))

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

# BFS 함수 정의
def bfs(x, y):
  queue = deque()
  queue.append((x, y))
  # 큐가 빌 때까지 반복
  while queue:
    x, y = queue.popleft()
    # 현재 위치에서 상하좌우 4 방향 확인
    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))

문제 변형 - (0, 0)부터 (n - 1, m - 1)까지 갈 수 있는 경로의 수

입력 예시

7 7 1111111 1000001 1110111 0010100 0011110 0010011 0111111

출력 예시

8

소스코드

• '경로의 수' ➡️ DFS 이용
• (0, 0)부터 탐색을 시작하여, (n - 1, m - 1)에 도착했을 때 경로의 수 + 1
• visited 리스트를 이용하여 현재 노드가 이전에 방문했는지를 체크함
  • (0, 0)은 방문하지 않도록 dfs 수행 전에 미리 visited 체크하기 
  • DFS에서 나올 때, 다른 DFS가 들어갈 수 있기 때문에 해제해주기

# 입력 받기
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]

visited = [[0] * m for _ in range(n)]

# DFS 함수 정의
answer = 0
def dfs(x, y):
  global answer
  # 우하단에 도착했을 경우 경로의 수 + 1
  if x == (n - 1) and y == (m - 1):
    answer += 1
  if x >= 0 and x < n and y >= 0 and y < m and graph[x][y] == 1:
    # 현재 노드의 상하좌우에 대해 탐색
    for i in range(4):
      nx = x + dx[i]
      ny = y + dy[i]
      # 탐색 범위 내에 있으며, 갈 수 있는 경로인 경우
      if nx >= 0 and nx < n and ny >= 0 and ny < m and visited[nx][ny] == 0:
         # 한번도 방문하지 않은 노드일 경우 방문 수행
        visited[nx][ny] = 1
        dfs(nx, ny)
        visited[nx][ny] = 0

visited[0][0] = 1
dfs(0, 0)
print(answer)