백준 3055 탈출 python

3055번: 탈출

문제

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

 

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

 

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

 

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

 

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

입력

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

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

출력

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

풀이

from collections import deque
n, m = map(int, input().split())
maps = [list(input()) for _ in range(n)]
# 영역의 크기와 현재 영역의 상태를 지도로 받아줍니다
dx = [1, -1, 0, 0]
dy = [0, 0, 1, -1]
# 4방향 이동을 설정해줍니다.
def bfs(a, b):
    q = deque()
    visited = [[0]* m for _ in range(n)]
    # 영역의 크기와 같은 방문 처리를 해 줄 리스트를 만들어줍니다.
    for i in range(n):
        for j in range(m):
            if maps[i][j] == '*':
                q.append((i, j))
                visited[i][j] = -1
                # 모든 영역을 확인하며 물이 차오르는 위치라면
                # 덱에 넣고 -1로 방문처리를 해줍니다.
    q.append((a, b))
    # 덱에 고슴도치의 위치를 담아줍니다.
    visited[a][b] = 1
    # 고슴도치의 현재 위치를 방문처리 하기위해 1로 바꾸어줍니다.
    while q:
        x, y = q.popleft()
        for i in range(4):
            nx = dx[i] + x
            ny = dy[i] + y
            if nx in range(n) and ny in range(m):
                # 덱에서 뺀 값이 4방위 이동하여도 영역의 안인지 확인해주고
                if maps[nx][ny] == 'D' and visited[x][y] > 0:
                    return visited[x][y]
                    # 비버의 소굴이고 고슴도치가 이동해온 것이라면 고슴도치가 이동하는데
                    # 든 시간이 얼마인지 되돌려줍니다. 시작위치를 1로 시작하였기 때문에
                    # visited[x][y]에 저장되어있는 값을 출력해줍니다.
                if maps[nx][ny] == '.':
                    if visited[nx][ny] == 0:
                    # 아니라면 지도에서 빈 공간인지 확인해주고
                    # 방문한 적이 있는지 확인해줍니다.
                        if visited[x][y] > 0:
                            visited[nx][ny] = visited[x][y] + 1
                            q.append((nx, ny))
                            # 고슴도치가 이동한 것이라면 시간이 1만큼 들었다고 표시해주고
                            # 덱에 좌표를 추가해주고
                        else:
                            visited[nx][ny] = -1
                            q.append((nx, ny))
                            # 물이 차오른것이라면 -1로 방문처리를 해주고 덱에 값을 추가해줍니다.
    return 'KAKTUS'
    # 덱이 비었다면 고슴도치가 탈출을 못 했다는 의미이므로 'KAKTUS'을 추가해줍니다.
for i in range(n):
    for j in range(m):
        if maps[i][j] == 'S':
            print(bfs(i, j))
            # 영역을 순회하면 고슴도치의 위치를 찾아 함수를 실행시켜 받는 결과를 출력해줍니다.