[Algorithm] BaekJoon : 3055. 탈출 by Python

엄희관·2021년 3월 6일
0

Algorithm

목록 보기
112/128
post-thumbnail

[문제 바로가기] https://www.acmicpc.net/problem/3055

📌문제 설명

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

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

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

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

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

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

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

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


💡 문제 풀이

문제의 지문이 길지만 정리해보면 '물(*)'과 '고슴도치(S)' 두 가지에 대해서 각각 BFS 탐색을 진행하면 된다.

이동 순서는 물의 이동이 고슴도치의 이동보다 우선하기 때문에(→ '다음 시간에 물이 찰 예정인 칸으로 고슴도치는 이동할 수 없다.')
물, 고슴도치 순으로 BFS탐색을 진행한다.

두 개의 BFS 탐색을 위해 두 개의 queue를 생성하였다.

  • queue_pet : 고슴도치
  • queue_water : 물

물은 단순히 이동만 체크하면 되기 때문에 (행, 열)자료를 큐에 담았고 고슴도치는 비버의 굴(D)에 도착하는 최소 시간을 구해야 하므로 (이동시간, 행, 열)자료를 큐에 담았다.

고슴도치를 이동시킬 때 이동 후 위치에는 'S'를 이동 전 위치에는 '.'로 수정하면서 진행한다.

만약 목적지(D)에 도착하면 정답을 출력할 'answer'변수에는 최소 시간값을 넣어주며 탐색을 마친다.

코드는 다음과 같다.

import sys
from collections import deque

d = [(-1, 0), (1, 0), (0, 1), (0, -1)]
R, C = map(int, input().split())
maps = [list(' '.join(map(str, sys.stdin.readline().split()))) for _ in range(R)]

def move_water():
    global R, C
    for _ in range(len(queue_water)):
        r, c = queue_water.popleft()
        for idx in range(4):
            nr = r + d[idx][0]
            nc = c + d[idx][1]
            if 0 <= nr < R and 0 <= nc < C:
                if maps[nr][nc] == '.':
                    maps[nr][nc] = '*'
                    queue_water.append((nr, nc))
                    visited[nr][nc] = 1

def move_pet(cnt):
    global answer
    for _ in range(len(queue_pet)):
        m, r, c = queue_pet.popleft()
        for idx in range(4):
            nr = r + d[idx][0]
            nc = c + d[idx][1]
            if 0 <= nr < R and 0 <= nc < C and not visited[nr][nc]:
                if maps[nr][nc] == '.':
                    queue_pet.append((cnt+1, nr, nc))
                    maps[nr][nc], maps[r][c] = 'S', '.'
                    visited[nr][nc] = 1
                if maps[nr][nc] == 'D':
                    answer = cnt+1

queue_pet = deque([])
queue_water = deque([])
visited = [[0] * C for _ in range(R)]
for r in range(R):
    for c in range(C):
        if maps[r][c] == 'S':
            queue_pet.append((0, r, c))
            visited[r][c] = 1
        if maps[r][c] == '*':
            queue_water.append((r, c))
            visited[r][c] = 1

answer = 'KAKTUS'
cnt = 0
while queue_pet: # 고슴도치가 이동이 가능할 때 까지 BFS 탐색
    move_water() # 1. 물 BFS
    move_pet(cnt) # 2. 고슴도치 BFS
    cnt += 1
    if answer != 'KAKTUS': break # answer값이 최소시간으로 변경되면 탐색을 마친다.
print(answer)

profile
허브

0개의 댓글