[파이썬] BOJ -3055 : 탈출

seanlion·2020년 12월 29일
0

알고리즘

목록 보기
8/11
post-thumbnail

링크

https://www.acmicpc.net/problem/3055

문제

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

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

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

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

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

입력

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

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

출력

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

풀이

- 이 문제는 BFS를 사용해 풀 수 있는 문제이다.

  • 물과 고슴도치를 상하좌우로 움직이면서 물을 확장해나가고, 고슴도치를 확장해나가다가 고슴도치가 D에 가면 종료되는 식으로 BFS를 구현하면 된다.

  • 고슴도치는 물이 찰 예정인 칸으로 이동할 수 없기 때문에 물이 고슴도치가 이동하는 것보다 한 타임 더 빠르게 움직여야 한다.

  • 그래서 큐에 물 좌표를 먼저 넣어주고 고슴도치 좌표를 넣어줬는데(그러면 알아서 물 먼저 움직이고, 그 다음 고슴도치가 움직일테니 물이 찰 예정인 칸을 피할 수 있게 될 줄 알았다.),
    실제로 코드를 돌려보니 물이 먼저 차서, 고슴도치가 갈 수 없는 상황이 만들어졌다.

  • 사실 고슴도치와 물이 동시에 움직이면 갈 수 있는 상황인데, 무조건 물이 먼저 움직이는 로직이라서 그랬던 것 같다.

  • 그래서 첫 생각을 조금 고쳐서 물을 먼저 큐에 넣긴하지만 마지막에 고슴도치를 큐 front에 넣어주고 고슴도치가 먼저 출발하도록 했다.

  • 어짜피 고슴도치는 BFS 돌면서 4방향으로 가고, 물이 고슴도치 영역을 뺏을 수 있기 때문에 고슴도치가 먼저 출발한다 해서 크게 상관은 없는 것 같다.

코드

import sys
from collections import deque
ROW,COL = map(int,sys.stdin.readline().split())
matrix = [list(sys.stdin.readline().strip()) for _ in range(ROW)]
visit_map = [[0] * COL for _ in range(ROW)]

queue = deque()
for row in range(ROW):
    for col in range(COL):
        if matrix[row][col] == '*':
            # 물 먼저 다 넣고 고슴도치 front에 넣기
            queue.append([row,col])
        elif matrix[row][col] == 'S':
            # matrix[row][col] = 0
            hodgedog = [row,col]
        elif matrix[row][col] == 'D':
            bv_r, bv_c = row,col

queue.appendleft(hodgedog)

dx = [-1,0,1,0] # 순서대로 좌, 상, 우, 하
dy = [0,1,0,-1]
def bfs(matrix,queue):
    flag = False
    while queue:
        x,y = queue.popleft()
        if flag: # 고슴도치가 D에 도착했으면
            break
        for i in range(4):
            next_x = x + dx[i]
            next_y = y + dy[i]
            # 고슴도치 - 바깥에 나가거나 돌, 물 지역은 가면 안됨.
            if 0<= next_x < ROW and 0 <= next_y < COL:
                # (물 입장에서)다음에 갈 지역이 비어있는 곳이고 
                # 비버 동굴이 아니면 물 영역을 확장한다.
                if matrix[x][y] == '*':
                    if matrix[next_x][next_y] == '.' or matrix[next_x][next_y] == 'S':
                        matrix[next_x][next_y] = '*'
                        queue.append([next_x,next_y])
                # (고슴이 입장에서) 다음에 갈 지역이 비어있고, 
                # 물이 아니고 바위가 아니고, 
                # 맵 상 이미 방문한 곳이 아니면 visit map에 이동 거리를 찍는다.
                elif matrix[x][y] == 'S':
  
                    if matrix[next_x][next_y] == '.':
                        matrix[next_x][next_y] = 'S'
                        visit_map[next_x][next_y] = visit_map[x][y] +1
                        queue.append([next_x, next_y])
                    elif matrix[next_x][next_y] == 'D': 
                    #고슴도치가 D에 도착할 수 있으면
                        flag = True
                        visit_map[next_x][next_y] = visit_map[x][y]+1
                        break
bfs(matrix,queue)
if visit_map[bv_r][bv_c] == 0:
    print('KAKTUS')
else:
    print(visit_map[bv_r][bv_c])
profile
소프트웨어 엔지니어링을 연마하고자 합니다.

0개의 댓글