[백준] 2589번 보물섬 파이썬

문제

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

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

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

입력

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

출력

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

예제 입력 1

5 7
WLLWWWL
LLLWLLL
LWLWLWW
LWLWLLL
WLLWLWW

예제 출력 1

8

제출답안

from collections import deque


def bfs(x, y):
    distance = 0
    q = deque()
    q.append((x, y, 0))
    visit[x][y] = 1

    while q:
        x, y, d = q.popleft()
        for k in range(4):
            nx = x + dx[k]
            ny = y + dy[k]
            if nx < 0 or ny < 0 or nx >= a or ny >= b:
                continue
            if graph[nx][ny] == 'L' and visit[nx][ny] == 0:
                q.append((nx, ny, d + 1))
                visit[nx][ny] = 1
                distance = d + 1

    return distance


dx = [0, 0, -1, 1]
dy = [-1, 1, 0, 0]
a, b = map(int, input().split())
graph = [input() for _ in range(a)]
max_dis = 0

for i in range(a):
    for j in range(b):
        if graph[i][j] == 'L':
            visit = [[0] * b for _ in range(a)]
            max_dis = max(max_dis, bfs(i, j))

print(max_dis)

1. 문제에서 가장 먼 곳에 보물을 뒀다고 하니 그게 보물이 묻힌 곳이고 최단거리이다.
2. 그러니 bfs로 거리를 구해서 최장거리를 찾으면 된다.
   1. 그래프 구조를 2중그래프로 우선 저장한다.
   2. 그리고 q에 저장할 때 누적 거리를 저장해준다.
3. 마지막 for구문에서 최장거리를 print하면 끝이다.