[백준] 3055. BFS_탈출 - Python

문제

사악한 암흑의 군주 이민혁은 드디어 마법 구슬을 손에 넣었고, 그 능력을 실험해보기 위해 근처의 티떱숲에 홍수를 일으키려고 한다. 이 숲에는 고슴도치가 한 마리 살고 있다. 고슴도치는 제일 친한 친구인 비버의 굴로 가능한 빨리 도망가 홍수를 피하려고 한다.

티떱숲의 지도는 R행 C열로 이루어져 있다. 비어있는 곳은 '.'로 표시되어 있고, 물이 차있는 지역은 '*', 돌은 'X'로 표시되어 있다. 비버의 굴은 'D'로, 고슴도치의 위치는 'S'로 나타내어져 있다.

매 분마다 고슴도치는 현재 있는 칸과 인접한 네 칸 중 하나로 이동할 수 있다. (위, 아래, 오른쪽, 왼쪽) 물도 매 분마다 비어있는 칸으로 확장한다. 물이 있는 칸과 인접해있는 비어있는 칸(적어도 한 변을 공유)은 물이 차게 된다. 물과 고슴도치는 돌을 통과할 수 없다. 또, 고슴도치는 물로 차있는 구역으로 이동할 수 없고, 물도 비버의 소굴로 이동할 수 없다.

티떱숲의 지도가 주어졌을 때, 고슴도치가 안전하게 비버의 굴로 이동하기 위해 필요한 최소 시간을 구하는 프로그램을 작성하시오.

고슴도치는 물이 찰 예정인 칸으로 이동할 수 없다. 즉, 다음 시간에 물이 찰 예정인 칸으로 고슴도치는 이동할 수 없다. 이동할 수 있으면 고슴도치가 물에 빠지기 때문이다.

입력

첫째 줄에 50보다 작거나 같은 자연수 R과 C가 주어진다.

다음 R개 줄에는 티떱숲의 지도가 주어지며, 문제에서 설명한 문자만 주어진다. 'D'와 'S'는 하나씩만 주어진다.

출력

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

 

-----------------------------------------------------------------------------------------------------------------------------------

 

★ 시작하기 전에

* 물이 퍼진다는 조건 만 없다면 기존 BFS 문제, DFS 문제로 풀 수 있다. 다만 시간에 따라 가만히 있던 board data가 이번 문제는 바뀐다는 제약 조건이 있다. 원래 풀던 BFS 문제 틀을 먼저 구성한 뒤, 어떤 시간에 board data를 바꾸어 줄까 생각했고 q에 넣어주는 시간 변수에 따라 board를 바꾸는 spread함수를 구현하여 풀었다.

import sys

input = sys.stdin.readline

n, m = map(int, input().split())
board = [list(map(str,input().strip())) for _ in range(n) ]

# q는 고슴도치가 이동하는 위치 및 시간을 담는다
# w는 확산될 홍수가 난 위치를 담는다.
q =[]
w = []
visited = [[0 for _ in range(m)] for _ in range(n)]
for i in range(n):
    for j in range(m):
        if board[i][j] == 'S':
            q.append((i, j, 0))
            visited[i][j] = 1
        elif board[i][j] == '*':
            w.append((i, j))
dr = [1, -1, 0, 0]
dc = [0, 0, 1, -1]

# 물이 퍼질때 board판을 바꾸는 함수
def spread():
    global w
    cnt = len(w)
    while cnt > 0:
        nowW = w.pop(0)
        for arr in range(4):
            nextRw = nowW[0] + dr[arr]
            nextCw = nowW[1] + dc[arr]
            if 0 <= nextRw < n and 0 <= nextCw < m and board[nextRw][nextCw] == '.':
                board[nextRw][nextCw] = '*'
                w.append((nextRw, nextCw))
        cnt -= 1
std = 0
ans = 'KAKTUS'
isGet = False

# 기본적인 BFS 틀로 시작한다.
while q:
    if isGet:
        break
    now = q.pop(0)
    
    # 이 부분이 기존 BFS와 차이라고 본다. 시간에 따라서만 물이 퍼져야 하기 때문에 아래 조건을 걸어준다.
    if std == now[2]:
        spread()
        std += 1
    for ar in range(4):
        nextR = now[0] + dr[ar]
        nextC = now[1] + dc[ar]
        if 0<=nextR<n and 0<=nextC<m and visited[nextR][nextC] == 0 and board[nextR][nextC] == 'D':
            ans = now[2] + 1
            isGet = True
            break
        if 0<=nextR<n and 0<=nextC<m and visited[nextR][nextC] == 0 and board[nextR][nextC] == '.':
            visited[nextR][nextC] =1
            q.append((nextR, nextC, now[2]+1))

print(ans)

처음에는 어떻게 풀지 어떻게 풀지... 고민이 되지만 그래도 간단한 방법으로 해결할 수 있었다. 이런 문제가 골드 5라고 생각한다면, 문제의 난이도는 푸는 개념에 따라 나뉘는 것이라고 생각이 든다.

더 윗 개념들은 이해하기 어렵겠지만 알고 나면 문제에 적용하여 풀 수 있지 않을....까?