[Python/BOJ] 2589. 보물섬

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

문제

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

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

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

입력

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

출력

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

풀이

Code

import sys
from collections import deque
input = sys.stdin.readline

# 1. bfs 함수 정의
def bfs(s_x, s_y) :
    # 1-1. 큐 생성 후 현재 위치 삽입
    queue = deque([(s_x, s_y)])
    # 1-2. 방문 여부 리스트 생성
    visited = [[0 for _ in range(m)] for _ in range(n)]
    # 1-3. 최대 길이 변수 생성
    max_len = 0
    # 1-4.
    while queue :
        # 1-4-1. 위치 인덱스 반환
        x, y = queue.popleft()
        # 1-4-2.
        for dir_x, dir_y in [(-1, 0), (0, 1), (1, 0), (0, -1)] :
            # 다음 위치 설정
            nx, ny = x + dir_x, y + dir_y
            # 예외 처리
            if nx < 0 or nx >= n or ny < 0 or ny >= m or (nx, ny) == (s_x, s_y) or graph[nx][ny] == "W" : continue
            # 현재 위치에 방문하지 않은 경우
            if not visited[nx][ny] :
                # 방문 처리
                visited[nx][ny] = visited[x][y] + 1
                # 최대 길이 업데이트
                max_len = max(max_len, visited[nx][ny])
                # 큐에 다음 위치 삽입
                queue.append((nx, ny))
    # 1-5. 최대 길이 반환
    return max_len

n, m = map(int, input().split())
graph = [list(input().strip()) for _ in range(n)]

# 2. 최단 거리 변수 생성
ans = -float("INF")
# 3.
for x in range(n) :
    for y in range(m) :
        # 3-1. 현재 위치가 육지인 경우
        if graph[x][y] == "L" :
            # 최단 거리 업데이트
            ans = max(ans, bfs(x, y))
# 4. 결과 출력
print(ans)