BFS : 탈출(백준 3055번)

▶ 문제 : https://www.acmicpc.net/problem/3055

3055번: 탈출

##### 문제 #####
# 티떱숲의 지도는 R행 C열로 이루어져 있다. 
# 비어있는 곳은 '.'로 표시되어 있고, 
# 물이 차있는 지역은 '*', 돌은 'X'로 표시되어 있다. 
# 비버의 굴은 'D'로, 고슴도치의 위치는 'S'로 나타내어져 있다.
# 매 분마다 고슴도치는 현재 있는 칸과 인접한 네 칸 중 하나로 이동할 수 있다. (위, 아래, 오른쪽, 왼쪽) 
# 물도 매 분마다 비어있는 칸으로 확장한다. 
# 물이 있는 칸과 인접해있는 비어있는 칸(적어도 한 변을 공유)은 물이 차게 된다. 
# 물과 고슴도치는 돌을 통과할 수 없다. 
# 또, 고슴도치는 물로 차있는 구역으로 이동할 수 없고, 물도 비버의 소굴로 이동할 수 없다.
# 티떱숲의 지도가 주어졌을 때, 
# 고슴도치가 안전하게 비버의 굴로 이동하기 위해 필요한 최소 시간을 구하는 프로그램을 작성하시오.
# 고슴도치는 물이 찰 예정인 칸으로 이동할 수 없다. 
# 즉, 다음 시간에 물이 찰 예정인 칸으로 고슴도치는 이동할 수 없다. 
# 이동할 수 있으면 고슴도치가 물에 빠지기 때문이다. 

##### 입력 #####
# 첫째 줄에 50보다 작거나 같은 자연수 R과 C가 주어진다.
# 다음 R개 줄에는 티떱숲의 지도가 주어지며, 문제에서 설명한 문자만 주어진다. 
# 'D'와 'S'는 하나씩만 주어진다.

##### 출력 #####
# 첫째 줄에 고슴도치가 비버의 굴로 이동할 수 있는 가장 빠른 시간을 출력한다. 
# 만약, 안전하게 비버의 굴로 이동할 수 없다면, "KAKTUS"를 출력한다.

 

▶ 접근 방법

고슴도치위 위치와 물의 위치를 어떻게 갱신해줄지가 관건이었다. 이를 해결하기 위해 큐를 이용했다.
초기 고슴도치의 위치와 물의 위치를 큐에 [고슴도치 -> 물] 순서로 입력한다.
이런 순서로 초기 값을 입력하는 이유는
이후 고슴도치의 위치와 물의 위치를 순서대로 꺼내고 입력하는 과정에서 위치 정보를 갱신할 텐데,
뒤에 나오는 물이 앞에 나온 고슴도치의 위치를 덮어씀으로써
다음 순서의 고슴도치가 해당 위치로 이동하는 것을 막기 위함이다. 

 

▶ 풀이 코드

import sys
from collections import deque

# 숲의 크기를 입력 값으로 받는다.
R, C = map(int, sys.stdin.readline().split())

# 숲의 정보를 입력 값으로 받는다.
forest_info = [list(sys.stdin.readline().rstrip()) for _ in range(R)]
# for forest in forest_info:
#     print(forest)

# 시작 전 이동 시간을 0으로 초기화 한다.
time_info = [[0] * C for _ in range(R)]
# for time in time_info:
#     print(time)

# 고슴도치와 물의 위치를 넣어 줄 queue를 하나 만들어준다.
queue = deque()

for row in range(R):
    for col in range(C):
        # forest_info에서 초기 고슴도치의 위치를 확인해서 queue에 넣어준다.
        if forest_info[row][col] == 'S':
            queue.append((row, col))
        # 비버의 굴 위치는 target_r, target_c라는 변수에 넣어준다.
        elif forest_info[row][col] == 'D':
            target_r, target_c = row, col

for row in range(R):
    for col in range(C):
        # forest_info에서 초기 물의 위치를 확인해서 queue에 넣어준다.
        if forest_info[row][col] == '*':
            queue.append((row, col))


def BFS(target_r, target_c):
    while queue:
        row, col = queue.popleft()
        # 고슴도치의 위치가 비버의 굴 위치와 같다면 그때의 시간을 반환한다.
        if forest_info[target_r][target_c] == 'S':
            return time_info[target_r][target_c]
            
        for dir_r, dir_c in (-1, 0), (1, 0), (0, -1), (0, 1):
            new_r, new_c = row + dir_r, col + dir_c

            # 숲의 크기를 벗어나지 않는 범위에서
            if 0 <= new_r < R and 0 <= new_c < C:
                # 고슴도치가 이동할 차례에 새롭게 이동하려는 곳이 빈 곳이거나 비버의 굴이라면
                if forest_info[row][col] == 'S' and (forest_info[new_r][new_c] == '.' or forest_info[new_r][new_c] == 'D'):
                    # 새롭게 이동하는 곳을 고슴도치의 위치로 갱신해주고, 시간 정보 역시 1 추가한다.
                    forest_info[new_r][new_c] = 'S'
                    time_info[new_r][new_c] = time_info[row][col] + 1
                    queue.append((new_r, new_c))
                # 물이 이동할 차례에 새롭게 이동하려는 곳이 빈 곳이거나 고슴도치의 위치라면
                elif forest_info[row][col] == '*' and (forest_info[new_r][new_c] == '.' or forest_info[new_r][new_c] == 'S'):
                    # 새롭게 이동하는 곳을 물의 위치로 갱신해준다.
                    forest_info[new_r][new_c] = '*'
                    queue.append((new_r, new_c))
    return "KAKTUS"

print(BFS(target_r,target_c))

 

▶ 관련 링크

2021.11.18 - [알고리즘] - BFS(Breadth First Search, 너비 우선 검색)이란?

2021.11.19 - [알고리즘] - 그래프(Graph)란?

2021.11.19 - [알고리즘] - BFS : 미로 탐색(백준 2178번)

2021.11.19 - [알고리즘] - BFS : 특정 거리의 도시 찾기(백준 18352번)

2021.11.19 - [알고리즘] - 다익스트라 알고리즘(Dijkstra Algorithm)이란?

2021.11.19 - [알고리즘] - BFS : 최소비용 구하기(백준 1916번)

2021.11.20 - [알고리즘] - BFS : 미로만들기(백준 2665번)