[백준] # 2589. 보물섬 (파이썬)

문제 링크

 

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

2589번: 보물섬

 

문제 내용

 

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

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

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

 

입력

 

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

 

출력

 

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

 

IDEA

 

육지 모두를 시작 지점으로 하는 BFS를 실행했다. 즉 육지의 개수만큼 BFS함수를 돌리는 것이다. 매우 비효율적이지만 어쨌든 풀렸다.. 파이썬으로 하면 시간 초과가 난다.

 

CODE

 

# 2589. 보물섬 - 골드5
# 무작정 bfs 최대 2500번하기!
# pypy3로 하니 통과!
from collections import deque
dx,dy = (1,-1,0,0),(0,0,1,-1)
def bfs(a,b):
    q = deque([(a,b,0)])
    visited = [[0]*c for _ in range(r)]
    visited[a][b] = 1
    time = 0
    while q:
        x,y,distance = q.popleft()
        time = max(time,distance)
        for i in range(4):
            nx = x + dx[i]
            ny = y + dy[i]
            if 0 <= nx < r and 0 <= ny < c:
                if arr[nx][ny] == "L" and visited[nx][ny] != 1:
                    visited[nx][ny] = 1
                    q.append((nx,ny,distance+1))
    return time

r,c = map(int,input().split())
land = []
arr = []
t = 0
for i in range(r):
    input_list = list(input().strip())
    arr.append(input_list)
    for j in range(c):
        if input_list[j] == "L":
            land.append((i,j))
for x,y in land:
    t = max(t,bfs(x,y))
print(t)

 

사용 알고리즘

 

BFS

 

풀이 평

 

무작정 BFS 하는 것보다 효율적인 게 있을까 생각해봤지만 생각이 나지 않아서 그냥 단순하게 최대 2500번의 BFS를 해서 pypy3로 제출했는데 통과했다.