[맛있는물회] [파이썬] <백준 알고리즘> 3055번 "탈출"

문제 조건

---

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

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

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

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

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

 

Input

---

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

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

 

Output

---

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

 

생각한 아이디어

---

재밌는 BFS문제~

단순히 BFS로 경로를 찾는 것이 아니라, 시간이 지날때마다 맵의 구성요소 (물의 위치)들이 바뀌는 시뮬레이션적 요소도 포함해주어야한다.

 

그래서 나는 BFS 함수 안에 모든 것들을 때려넣었다.

 

while 문이 시작하면서 물이 차는 반복문을 수행한다.

물이 찬 새로운 MAP은 tmp에 저장한다.

왜냐면 물이 차면서 MAP이 변경되면 그 시간에 변경된 물에 대해서 다시 반복문을 돌 것이기 때문에 Original MAP은 보존하고 tmp 에 저장후 물이 다 차고 나면 tmp를 map에 넣어준다.

 

그리고 고슴도치는 움직인다.

고슴도치는 물, X(벽)을 통과하지 못한다.

이를 염두에 두고 bfs를 수행한다.

여기서 키포인트는 queue이다.

 

한번에 시뮬레이션 까지 돌리다 보니 하나의 queue로는 불가능하다.

따라서 다음 위치에 올 값들은 next_q에 넣어준다.

그리고 마지막에 next_q값을 다시 queue에 넣어주고 초기화한다.

그렇게되면 현재 움직이는 값을 다 pop해주고 다음에 올 값을 가진 후 time이 1 증가하면 다시 수행할 수 있다.

 

마지막으로 고슴도치가 도착하거나, 도착하지 못하는 경우를 queue가 비어있는지로 판단하여 time을 출력한다.

 

 

소스코드

#3055 탈출

import sys
from collections import deque
import copy
input=sys.stdin.readline

dy=[-1,1,0,0]
dx=[0,0,-1,1]

def bfs(end_y,end_x,start_y,start_x,MAP,row,col):
   queue,next_q = deque(),deque()
   queue.append((start_y,start_x))
   visited=[[0 for _ in range(col)]for _ in range(row)]
   visited[start_y][start_x]=1
   time=0
   while 1:
      tmp=copy.deepcopy(MAP) #물이 넘치는 부분
      for y in range(0,row):
         for x in range(0,col):
            if MAP[y][x]=='*':
               for dir in range(0,4):
                  ny,nx=y+dy[dir],x+dx[dir]
                  #물은 비어있는 곳만 찬다.
                  if row>ny>=0 and col>nx>=0 and MAP[ny][nx]=='.':
                     tmp[ny][nx]='*'
      MAP=tmp
      if not queue: return 'KAKTUS'

      while queue:
         y,x=queue.popleft()
         if (y,x) == (end_y,end_x):return time
         for dir in range(0,4):
            ny,nx=y+dy[dir],x+dx[dir]
            #고슴도치는 물, 돌 통과 X
            if row>ny>=0 and col>nx>=0 and visited[ny][nx]==0 and MAP[ny][nx]!='*' and MAP[ny][nx]!='X':
               visited[ny][nx]=1
               next_q.append((ny,nx))
      queue.extend(next_q)
      next_q=[]
      time+=1

if __name__ == "__main__":
   row,col = map(int,input().split())
   MAP=[list(str(input().rstrip())) for _ in range(row)]
   for y in range(0,row):
      for x in range(0,col):
         if MAP[y][x] == 'S':
            start_y,start_x=y,x
         elif MAP[y][x] == 'D':
            end_y,end_x=y,x
   ans = bfs(end_y,end_x,start_y,start_x,MAP,row,col)
   print(ans)

 

*파이썬 문법 정리