문제 바로가기
문제:
당신은 상범 빌딩에 갇히고 말았다. 여기서 탈출하는 가장 빠른 길은 무엇일까? 상범 빌딩은 각 변의 길이가 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!
풀이:
3차원 BFS 문제이다
import sys
from collections import deque
sys.stdin = open('/Users/song/Desktop/Python/Python/h.txt', 'r')
move = [(0, 0, 1), (0, 0, -1), (0, 1, 0), (0, -1, 0), (1, 0, 0), (-1, 0, 0)]
while True:
l, r, c = map(int, sys.stdin.readline().rstrip().split(" "))
if l == r == c == 0:
break
graph = []
for _ in range(l):
graph.append([list(input().strip()) for _ in range(r)])
temp = input()
visited = [[[False] * c for _ in range(r)] for _ in range(l)]
q = deque()
for i in range(l):
for j in range(r):
for k in range(c):
if graph[i][j][k] == 'S':
start = (i, j, k, 0)
visited[i][j][k] = True
if graph[i][j][k] == 'E':
end = (i, j, k)
q.append(start)
while q:
floor, x, y, count = q.popleft()
if (floor, x, y) == end:
print(f'Escaped in {count} minute(s).')
break
for df, dx, dy in move:
nf = df + floor
nx = dx + x
ny = dy + y
if 0 <= nf < l and 0 <= nx < r and 0 <= ny < c and not visited[nf][nx][ny]:
if graph[nf][nx][ny] == "." or graph[nf][nx][ny] == "E":
visited[nf][nx][ny] = True
q.append((nf, nx, ny, count + 1))
else:
print("Trapped!")
처음에 3차원리스트를 컴프리핸션으로 하려다가 시간을 많이 사용했다.
만드는데 시간이 걸렸다.
댓글