[Python] 백준 6593번 - 상범 빌딩 (골드 5)

혼자 힘으로 풀었는가? O

알고리즘 분류
 - BFS

 

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

6593번: 상범 빌딩

 

문제

당신은 상범 빌딩에 갇히고 말았다. 여기서 탈출하는 가장 빠른 길은 무엇일까? 상범 빌딩은 각 변의 길이가 1인 정육면체(단위 정육면체)로 이루어져있다. 각 정육면체는 금으로 이루어져 있어 지나갈 수 없거나, 비어있어서 지나갈 수 있게 되어있다. 당신은 각 칸에서 인접한 6개의 칸(동,서,남,북,상,하)으로 1분의 시간을 들여 이동할 수 있다. 즉, 대각선으로 이동하는 것은 불가능하다. 그리고 상범 빌딩의 바깥면도 모두 금으로 막혀있어 출구를 통해서만 탈출할 수 있다.

당신은 상범 빌딩을 탈출할 수 있을까? 만약 그렇다면 얼마나 걸릴까?

입력

입력은 여러 개의 테스트 케이스로 이루어지며, 각 테스트 케이스는 세 개의 정수 L, R, C로 시작한다. L(1 ≤ L ≤ 30)은 상범 빌딩의 층 수이다. R(1 ≤ R ≤ 30)과 C(1 ≤ C ≤ 30)는 상범 빌딩의 한 층의 행과 열의 개수를 나타낸다.

그 다음 각 줄이 C개의 문자로 이루어진 R개의 행이 L번 주어진다. 각 문자는 상범 빌딩의 한 칸을 나타낸다. 금으로 막혀있어 지나갈 수 없는 칸은 '#'으로 표현되고, 비어있는 칸은 '.'으로 표현된다. 당신의 시작 지점은 'S'로 표현되고, 탈출할 수 있는 출구는 'E'로 표현된다. 각 층 사이에는 빈 줄이 있으며, 입력의 끝은 L, R, C가 모두 0으로 표현된다. 시작 지점과 출구는 항상 하나만 있다.

출력

각 빌딩에 대해 한 줄씩 답을 출력한다. 만약 당신이 탈출할 수 있다면 다음과 같이 출력한다.

Escaped in x minute(s).

여기서 x는 상범 빌딩을 탈출하는 데에 필요한 최단 시간이다.

만일 탈출이 불가능하다면, 다음과 같이 출력한다.

Trapped!

 

 

---

 

흔한 BFS 문제이며, 2차원 배열에서 3차원 배열로 넘어온것 뿐이다.

 

import sys
input = sys.stdin.readline

from collections import deque

while True:
    l, r, c = map(int, input().split())
    if l == 0 and r == 0 and c == 0:
        break
    data = [ [] for _ in range(l)]
    visited = [[[0 for _ in range(c)] for _ in range(r)] for _ in range(l)]
    q = deque()
    
    flag = False
    for i in range(l):
        for j in range(r):
            line = input().rstrip()
            data[i].append(line)
            if flag == False:
                for k in range(c):
                    if list(line)[k] == 'S':
                        q.append([i, j, k, 0])
                        flag = True
                        break
        input()
        
    dz = [-1, 1]
    dx = [-1, 0, 1, 0]
    dy = [0, 1, 0, -1]
    
    find = False
    while q:
        z, x, y, cnt = q.popleft()
        visited[z][x][y] = 1
        
        if data[z][x][y] == 'E':
            print(f'Escaped in {cnt} minute(s).')
            find = True
            break
        
        for i in range(2):
            nz = z + dz[i]
            if 0 <= nz < l:
                if visited[nz][x][y] == 0 and data[nz][x][y] != '#':
                    if data[z][x][y] == 'E':
                        print(f'Escaped in {cnt+1} minute(s).')
                        find = True
                        break
                    q.append([nz, x, y, cnt+1])
                    visited[nz][x][y] = 1
                    
        for i in range(4):
            nx = x + dx[i]
            ny = y + dy[i]
            if 0 <= nx < r and 0 <= ny < c:
                if visited[z][nx][ny] == 0 and data[z][nx][ny] != '#':
                    if data[z][x][y] == 'E':
                        print(f'Escaped in {cnt+1} minute(s).')
                        find = True
                        break
                    q.append([z, nx, ny, cnt+1])
                    visited[z][nx][ny] = 1
        if find:
            break
                    
    if find == False:
        print("Trapped!")