문제:
당신은 상범 빌딩에 갇히고 말았다. 여기서 탈출하는 가장 빠른 길은 무엇일까? 상범 빌딩은 각 변의 길이가 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!
풀이방법:
상하좌우로만 움직이는 일반 그래프 문제와 다르게 위, 아래의 개념이 추가된 문제다. 따라서 기존 2중 배열로 구성되던 visited 배열을 3중 배열로 바꿔서 풀도록 한다.
첫 시작점을 찾은 뒤에 BFS를 통해 움직일 수 없을 때까지 계속 움직이고, 만약 목적지에 도달하면 그 때의 시간을 반환하고, 그렇지 못하면 0을 반환하게 한다.
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
|
def bfs(start, visited):
queue = [start]
visited[start[0]][start[1]][start[2]] = 1
time = 1
while queue:
tmp = []
for q in queue:
z,x,y = q
for i in range(6):
nx = x + dx[i]
ny = y + dy[i]
nz = z + dz[i]
if 0<= nx < R and 0<= ny < C and 0<= nz < L and not visited[nz][nx][ny] and (building[nz][nx][ny]=='.' or building[nz][nx][ny]=='E'):
if building[nz][nx][ny]=='E':
return time
visited[nz][nx][ny] = 1
tmp.append([nz,nx,ny])
queue = tmp
time+=1
return 0
dx = [0,0,1,0,0,-1]
dy = [0,1,0,0,-1,0]
dz = [1,0,0,-1,0,0]
while True:
L, R, C = map(int,input().split())
if L==R==C==0:
break
building = []
visited = [[[0 for _ in range(C)] for _ in range(R)] for _ in range(L)]
for l in range(L):
floor = []
for r in range(R):
tmp = list(input())
if 'S' in tmp:
start_point = [l,r,tmp.index('S')]
floor.append(tmp)
input()
building.append(floor)
time = bfs(start_point,visited)
if time:
print(f'Escaped in {time} minute(s).')
else:
print("Trapped!")
|
cs |
문제링크:
'Algorithm > Python' 카테고리의 다른 글
[BOJ] 1019. 책 페이지 (0) | 2021.10.01 |
---|---|
[BOJ]3020. 개똥벌레 (0) | 2021.09.30 |
[BOJ]12904. A와 B (0) | 2021.09.28 |
[BOJ]1722. 순열의 순서 (0) | 2021.09.27 |
[BOJ]1351. 무한 수열 (0) | 2021.09.24 |