백준 온라인 저지, BFS / 2589번: 보물섬 (파이썬 / 백준 골드문제)

문제

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

문제 정의

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

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

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

입력

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

출력

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

예제 입력 1

5 7
WLLWWWL
LLLWLLL
LWLWLWW
LWLWLLL
WLLWLWW

예제 출력 1

8

접근 방법

- 모든 w에 대해 bfs를 탐색한 뒤, 가장 값이 높게 나온 것을 출력한다.
- 최대 연산 횟수: 50 x 50 x 2500 x 2 = 13,000,000회

코드

# https://www.acmicpc.net/problem/2589
# 접근 방법
# 모든 w에 대해 bfs를 탐색한 뒤, 가장 값이 높게 나온 것을 출력한다.
# 최대 연산 횟수: 50 x 50 x 2500 x 2 = 13,000,000회

def bfs(r, c):
    global time
    board = [x[:] for x in map_]
    queue = deque([])
    queue.append([r, c])
    board[r][c] = 0

    while queue:
        r_, c_ = queue.popleft()
        for dr, dc in [[1, 0], [-1, 0], [0, 1], [0, -1]]:
            if 0<=r_+dr<=row-1 and 0<=c_+dc<=col-1 and board[r_+dr][c_+dc] == 'L':
                board[r_+dr][c_+dc] = board[r_][c_] + 1
                queue.append([r_+dr, c_+dc])
        time = max(time, board[r_][c_])

from collections import deque
import sys
input = sys.stdin.readline
row, col = map(int, input().split())
map_ = [[x for x in input().rstrip()] for _ in range(row)]
time = 0

for r in range(row):
    for c in range(col):
        if map_[r][c] == 'L':
            bfs(r, c)
print(time)