성능 요약
메모리: 116604 KB, 시간: 160 ms
분류
너비 우선 탐색, 그래프 이론, 그래프 탐색
문제 설명
사악한 암흑의 군주 이민혁은 드디어 마법 구슬을 손에 넣었고, 그 능력을 실험해보기 위해 근처의 티떱숲에 홍수를 일으키려고 한다. 이 숲에는 고슴도치가 한 마리 살고 있다. 고슴도치는 제일 친한 친구인 비버의 굴로 가능한 빨리 도망가 홍수를 피하려고 한다.
티떱숲의 지도는 R행 C열로 이루어져 있다. 비어있는 곳은 '.'로 표시되어 있고, 물이 차있는 지역은 '*', 돌은 'X'로 표시되어 있다. 비버의 굴은 'D'로, 고슴도치의 위치는 'S'로 나타내어져 있다.
매 분마다 고슴도치는 현재 있는 칸과 인접한 네 칸 중 하나로 이동할 수 있다. (위, 아래, 오른쪽, 왼쪽) 물도 매 분마다 비어있는 칸으로 확장한다. 물이 있는 칸과 인접해있는 비어있는 칸(적어도 한 변을 공유)은 물이 차게 된다. 물과 고슴도치는 돌을 통과할 수 없다. 또, 고슴도치는 물로 차있는 구역으로 이동할 수 없고, 물도 비버의 소굴로 이동할 수 없다.
티떱숲의 지도가 주어졌을 때, 고슴도치가 안전하게 비버의 굴로 이동하기 위해 필요한 최소 시간을 구하는 프로그램을 작성하시오.
고슴도치는 물이 찰 예정인 칸으로 이동할 수 없다. 즉, 다음 시간에 물이 찰 예정인 칸으로 고슴도치는 이동할 수 없다. 이동할 수 있으면 고슴도치가 물에 빠지기 때문이다.
입력
첫째 줄에 50보다 작거나 같은 자연수 R과 C가 주어진다.
다음 R개 줄에는 티떱숲의 지도가 주어지며, 문제에서 설명한 문자만 주어진다. 'D'와 'S'는 하나씩만 주어진다.
출력
첫째 줄에 고슴도치가 비버의 굴로 이동할 수 있는 가장 빠른 시간을 출력한다. 만약, 안전하게 비버의 굴로 이동할 수 없다면, "KAKTUS"를 출력한다.
아이디어, 문제풀이
- 물이 찬 후에, 고슴도치가 이동한다.
- 한번씩 한번씩 진행할 수있도록 해야한다.
- 그 외는 기본 틀로 구현 할 수 있다.
TROUBLE SHOOTING
- 위에 기본틀로 구현 할 수 있다고 적어놨는데, 진짜 죽을뻔 했다. 일단, 이 문제는 한번씩 한번씩을 어떻게 구현할건지만 알면된다. 다른건 결국 전부다 똑같으니깐! 일단 내가 먼저 구현한 코드를 보면서 설명하겠다.
솔직히 잘했다고 생각했다. 근데, 틀렸다. 지금 보면 눈에 바로 들어오는걸 보니, 아직은 멀었다. 이 코드는 결국 물이 다 퍼진다음 고슴도치가 이동하는것과 다름이 없다. 생각은 잘 한것 같은데… 이런식의 알고리즘을 분명 다른 문제에서 사용할것 이라고 생각한다. 어쨌든, 틀린 코드다.while S_que: while W_que: a, b = W_que.popleft() for i in range(4): na = a + da[i] nb = b + db[i] if na<0 or na>= n or nb<0 or nb>= c: continue if graph[na][nb] == '.': W_que.append([na,nb]) graph[na][nb] = 'X' x, y = S_que.popleft() for i in range(4): nx = x + dx[i] ny = y + dy[i] if nx<0 or nx>= n or ny<0 or ny>= c: continue if graph[nx][ny] == 'X': continue if graph[nx][ny] == '.': W_que.append([nx,ny]) graph[nx][ny] = graph[x][y] + 1
- 내 머릿속 깊은곳에 존재했던 기억의 파편과, chat gpt의 도움을 받아 문제의 정답 코드를 구현했다.
마지막으로 움직이는 고슴도치를 기준으로 while문을 열어줘야하고, for문을 두번 활용해서 한칸씩, 즉, 한단계씩 이동할 수 있도록 설계했다. 그리고 여태까지 그냥 지나쳐가긴 했는데, 예외인 부분들을 마지막 return에 넣는것도 정형화 하는편이 좋을것 같다. 각 조건에 따라 출력값을 return하는 방식으로 구현하는것도 좋은데, 문제에서 예상치 못한 예외가 생기는 경우가 많았다. 물론 문제를 다시 리뷰할땐 하나하나 따져보며 지나가야 겠지만, 실전이라고 생각한다면 아닌 경우를 default로 가져가는게 좋을것 같다.def bfs(n, c): D = [] S_que = deque() W_que = deque() for i in range(n): for j in range(c): if graph[i][j] == 'S': S_que.append([i, j, 0]) if graph[i][j] == '*': W_que.append([i, j]) if graph[i][j] == 'D': D = [i, j] while S_que: for _ in range(len(W_que)): #코드 for _ in range(len(S_que)): #코드 return "KAKTUS"
코드
#https://www.acmicpc.net/problem/3055
#탈출
#3055
from collections import deque
import sys
input = sys.stdin.readline
n, c = map(int, input().split())
graph = []
for i in range(n):
a = list(map(str, input().strip()))
graph.append(a)
da = [1, -1, 0, 0]
db = [0, 0, 1, -1]
def bfs(n, c):
D = []
S_que = deque()
W_que = deque()
for i in range(n):
for j in range(c):
if graph[i][j] == 'S':
S_que.append([i, j, 0])
if graph[i][j] == '*':
W_que.append([i, j])
if graph[i][j] == 'D':
D = [i, j]
while S_que:
for _ in range(len(W_que)):
a, b = W_que.popleft()
for i in range(4):
na = a + da[i]
nb = b + db[i]
if na < 0 or na >= n or nb < 0 or nb >= c:
continue
if graph[na][nb] == '.':
W_que.append([na, nb])
graph[na][nb] = '*'
for _ in range(len(S_que)):
x, y, cnt = S_que.popleft()
if [x, y] == D:
return cnt
for i in range(4):
nx = x + da[i]
ny = y + db[i]
if nx < 0 or nx >= n or ny < 0 or ny >= c:
continue
if graph[nx][ny] == '*':
continue
if graph[nx][ny] == '.' or graph[nx][ny] == 'D':
S_que.append([nx, ny])
graph[nx][ny] = graph[x][y] + 1
return "KAKTUS"
print(bfs(n, c))