[BOJ] 2589 보물섬

https://www.acmicpc.net/problem/2589

2589번: 보물섬

보물섬

문제

보물섬 지도를 발견한 후크 선장은 보물을 찾아나섰다. 보물섬 지도는 아래 그림과 같이 직사각형 모양이며 여러 칸으로 나뉘어져 있다. 각 칸은 육지(L)나 바다(W)로 표시되어 있다. 이 지도에서 이동은 상하좌우로 이웃한 육지로만 가능하며, 한 칸 이동하는데 한 시간이 걸린다. 보물은 서로 간에 최단 거리로 이동하는데 있어 가장 긴 시간이 걸리는 육지 두 곳에 나뉘어 묻혀있다. 육지를 나타내는 두 곳 사이를 최단 거리로 이동하려면 같은 곳을 두 번 이상 지나가거나, 멀리 돌아가서는 안 된다.

예를 들어 위와 같이 지도가 주어졌다면 보물은 아래 표시된 두 곳에 묻혀 있게 되고, 이 둘 사이의 최단 거리로 이동하는 시간은 8시간이 된다.

보물 지도가 주어질 때, 보물이 묻혀 있는 두 곳 간의 최단 거리로 이동하는 시간을 구하는 프로그램을 작성하시오.

입력

첫째 줄에는 보물 지도의 세로의 크기와 가로의 크기가 빈칸을 사이에 두고 주어진다. 이어 L과 W로 표시된 보물 지도가 아래의 예와 같이 주어지며, 각 문자 사이에는 빈 칸이 없다. 보물 지도의 가로, 세로의 크기는 각각 50이하이다.

출력

첫째 줄에 보물이 묻혀 있는 두 곳 사이를 최단 거리로 이동하는 시간을 출력한다.

풀이

전형적인 그래프 이론 문제이다. 최단거리 문제가 나오면 그냥 BFS부터 적용하고 보는 성격이라 이번에도 마찬가지였다. 그냥 간단하게 n, m이 50보다 작으므로 모든 육지를 스캔할 때마다 BFS로 가장 먼 거리에 있는 육지로 가는 최단거리를 찾았다. 물론 더 효율적으로 풀 방법이 분명히 있을 것 같았다. 쨋든 python으로는 시간초과가 나고 pypy3으로는 통과했다.

import sys
from collections import deque
input = lambda: sys.stdin.readline().rstrip()
n, m = map(int, input().split())
treasure_map = [input() for _ in range(n)]
dx = [1, -1, 0, 0]
dy = [0, 0, 1, -1]
total_max_distance = 0
for i in range(n):
  for j in range(m):
    if treasure_map[i][j] == 'L':
      q = deque()
      q.append((i, j, 0))
      visited = [[False for _ in range(m)]for _ in range(n)]
      visited[i][j] = True
      max_dist = 0
      while q:
        x, y, dist = q.popleft()
        if max_dist < dist:
          max_dist = dist
        for k in range(4):
          nx = x + dx[k]
          ny = y + dy[k]
          if 0 <= nx < n and 0 <= ny < m:
            if not visited[nx][ny]:
              if treasure_map[nx][ny] == 'L':
                q.append((nx, ny, dist+1))
                visited[nx][ny] = True
      if total_max_distance < max_dist:
        total_max_distance = max_dist
print(total_max_distance)