2589 - 보물섬
문제
보물섬 지도를 발견한 후크 선장은 보물을 찾아나섰다. 보물섬 지도는 아래 그림과 같이 직사각형 모양이며 여러 칸으로 나뉘어져 있다. 각 칸은 육지(L)나 바다(W)로 표시되어 있다. 이 지도에서 이동은 상하좌우로 이웃한 육지로만 가능하며, 한 칸 이동하는데 한 시간이 걸린다. 보물은 서로 간에 최단 거리로 이동하는데 있어 가장 긴 시간이 걸리는 육지 두 곳에 나뉘어 묻혀있다. 육지를 나타내는 두 곳 사이를 최단 거리로 이동하려면 같은 곳을 두 번 이상 지나가거나, 멀리 돌아가서는 안 된다.
예를 들어 위와 같이 지도가 주어졌다면 보물은 아래 표시된 두 곳에 묻혀 있게 되고, 이 둘 사이의 최단 거리로 이동하는 시간은 8시간이 된다.
보물 지도가 주어질 때, 보물이 묻혀 있는 두 곳 간의 최단 거리로 이동하는 시간을 구하는 프로그램을 작성하시오.
코드
from collections import deque
def bfs(i, j):
q = deque()
q.append((i, j))
visited = [[0] * W for _ in range(L)]
visited[i][j] = 1
result = 0
while q:
x, y = q.popleft()
for i in range(4):
nx = x + dx[i]
ny = y + dy[i]
if 0 <= nx < L and 0 <= ny < W and visited[nx][ny] == 0 and graph[nx][ny] == 'L':
visited[nx][ny] = visited[x][y] + 1
result = max(result, visited[nx][ny])
q.append((nx, ny))
return result - 1
L, W = map(int, input().split())
graph = [list(input()) for _ in range(L)]
dx = [-1, 1, 0, 0]
dy = [0, 0, -1, 1]
start = []
answer = 0
for i in range(L):
for j in range(W):
if graph[i][j] == 'L':
answer = max(answer, bfs(i, j))
print(answer)한마디
처음에 find_start() 라는 함수를 만들어 시작 좌표들을 구해 BFS로 탐색했는데 계속 틀리다고 나왔다. 브루트포스로 모든 경우를 탐색하니까 통과했는데, 왜 시작 좌표들로만 탐색하면 안되는지 아직도 모르겠다.
