[백준] 3055_탈출 python

김동완·2022년 4월 30일
0

알고리즘

목록 보기
35/55
post-thumbnail

탈출

문제

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

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

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

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

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

입력

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

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

출력

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

해결방법

  • 최단거리니까 BFS로 해결한다.

  • 고슴도치와 물이 움직이니까 둘다 큐를 돌린다.

  • 고슴도치를 하나라고 생각하지 말고 갈 수 있는 방향에 다 갈 수 있게 하고 물이 차면 물이 덮게한다.

  • 따라서, 고슴도치를 먼저 돌리고 물이 담긴 큐를 번갈아가면서 돌린다

    ​ => 오답

  • 고슴도치 큐와 물큐를 번갈아가면서 돌렸더니 실패해서 고슴도치 큐와 물큐를 한 큐에 담아서 고슴도치를 먼저 담았더니 해결되었다.

# empty = . water = *, rock = 'X', biber = D go = 'S'

from collections import deque
#방향 설정 
dr = [0,1,0,-1]
dc = [1,0,-1,0]

n,m = map(int,input().split())

#물이 범람한 곳 
visit_water = []
#맵 받아오기 
grid = [list(input()) for _ in range(n)]
#방문배열 
visited = [[0]*m for _ in range(n)]
#그리드 돌면서 
wq = []
for r in range(n) :
    for c in range(m) :
        #도착지점 
        if grid[r][c] == 'D' :
            er,ec = r,c 
        #시작지점 
        elif grid[r][c] == 'S' :
            sr,sc = r,c 
            visited[r][c] = 1 
        #물이 있는 곳과 돌이 있는 곳 
        elif grid[r][c] == '*' :
            wq.append((r,c))
        elif grid[r][c] =='X' :
            visited[r][c] = 1
#시작지점에서 돌리는 큐 
qs = [(sr,sc)]+wq
q = deque(qs)
ans = 'KAKTUS'

#물큐와 고슴도치 큐를 돌면서 
while q :

    r, c = q.popleft()

    for d in range(4) :
        nr = r+dr[d]
        nc = c+dc[d]
        if 0<=nr<n and 0<=nc<m  :
            if grid[r][c] == 'S' and not visited[nr][nc] and (grid[nr][nc] == '.' or grid[nr][nc] == 'D') :
                visited[nr][nc] = visited[r][c] + 1 
                grid[nr][nc] = 'S'
                q.append((nr,nc))
                if nr==er and nc == ec :
                    ans = visited[r][c] 
            elif grid[r][c] == '*' and (grid[nr][nc] == 'S' or grid[nr][nc] == '.') :
                grid[nr][nc] = '*'
                q.append((nr,nc))

print(ans)
profile
내가 공부한 내용들이 누군가에게 도움이 될지 몰라서 쓰는 벨로그

0개의 댓글