[백준/Python] 6593 상범 빌딩

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으로 표현된다. 시작 지점과 출구는 항상 하나만 있다.

 

예제 입력

3 4 5
S....
.###.
.##..
###.#

#####
#####
##.##
##...

#####
#####
#.###
####E

1 3 3
S##
#E#
###

0 0 0

 

출력

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

Escaped in x minute(s).

 

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

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

Trapped!

 

예제 출력

Escaped in 11 minute(s).
Trapped!

 

문제 접근

그 동안 2차원 그래프 문제만 풀다가 처음으로 3차원 그래프 문제를 만났다. 좌표 이동 배열(dx, dy)를 항상 만들어서 사용한 덕분에 상하 이동 배열(dz)도 쉽게 만들어서 적용할 수 있었다. 

 

탈출할 수 있는 최단거리를 찾는 문제이므로 BFS 알고리즘을 사용하였다. 

 

정답은 맞췄지만 비효율적이라고 생각이 드는 코드들이 몇 군데 있었다. 추후 좀 더 효율적인 로직을 구상해 다시 풀어봐야겠다. 

 

최초 풀이

# 상범빌딩 탈출하기
# 동서남북상하 이동 가능. 1칸당 1분 소요.
# 금으로 막힌 칸은 지나갈 수 없음
# 출구로만 탈출 가능

import sys
input = sys.stdin.readline 
from collections import deque

dx = [1, 0, -1, 0, 0, 0]
dy = [0, 1, 0, -1, 0, 0]
dz = [0, 0, 0, 0, -1, 1]

def is_valid_coord(z, y, x):
  return 0 <= x < C and 0 <= y < R and 0 <= z < L 

def bfs(z, y, x):
  dq = deque()
  dq.append((z,y,x))
  chk[z][y][x] = True
  board[z][y][x] = 0

  while dq:
    z, y, x = dq.popleft()

    for k in range(6):
      nz = z + dz[k]
      ny = y + dy[k]
      nx = x + dx[k]

      if is_valid_coord(nz, ny, nx) and not chk[nz][ny][nx] and board[nz][ny][nx] != '#':
        
        if board[nz][ny][nx] == 'E': 
          print(f"Escaped in {board[z][y][x] + 1} minute(s).")
          return 

        dq.append((nz, ny, nx))
        chk[nz][ny][nx] = True
        board[nz][ny][nx] = board[z][y][x] + 1
        
  print("Trapped!")      

while True: 
  # L : 층 수, R : 행, C : 열  
  L, R, C = map(int, input().split())

  if L == 0 and R == 0 and C == 0:
    break 

  board = []
  chk = [[[False for _ in range(C)] for _ in range(R)] for _ in range(L)]
  for _ in range(L):
    board.append([list(input().rstrip()) for _ in range(R)])
    t = input()

  for z in range(L):
    for y in range(R):
      for x in range(C):
        if board[z][y][x] == 'S':
          bfs(z, y, x)