알고리즘 풀이/백준

[백준 2589] 보물섬

mhko411 2021. 6. 21. 17:25
728x90

문제

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

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

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

 

입력

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

 

출력

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


접근

처음에는 특정 육지에서 출발하여 가장 먼 육지를 찾고 이 육지부터 출발하여 거리가 가장 먼 곳을 찾았다. 입력 예시로 했을 때는 값이 올바르게 나왔지만 아직까지 틀린 이유를 찾지못했다.

 

그래서 완전탐색을 하고 거리를 구할 때는 BFS를 활용하였다.

 

구현

- 모든 육지에 대하여 BFS를 실행한다.

- BFS를 통해 현재 육지에서 가장 먼 거리를 반환하게 된다.

for y in range(N):
    for x in range(M):
        if board[y][x] == 'L':
            answer = max(answer, bfs(y, x))

 

- BFS를 통해 현재 위치에서 가장 멀리 있는 곳을 찾고

- 가장 멀리있는 곳까지의 거리를 반환한다.

def bfs(y, x):
    q = deque()
    dist = [[0 for _ in range(M)] for _ in range(N)]
    dist[y][x] = 1
    q.append((y, x))
    max_dist = -1
    dy = [1, -1, 0, 0]
    dx = [0, 0, 1, -1]
    while q:
        cy, cx = q.popleft()

        if dist[cy][cx] > max_dist:
            max_dist = dist[cy][cx]
        for d in range(4):
            ny = cy + dy[d]
            nx = cx + dx[d]
            if check_range(ny, nx) and board[ny][nx] == 'L':
                if dist[ny][nx] == 0:
                    dist[ny][nx] = dist[cy][cx] + 1
                    q.append((ny, nx))
    return max_dist - 1

전체 코드

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

def check_range(y, x):
    return (0 <= y < N) and (0 <= x < M)

def bfs(y, x):
    q = deque()
    dist = [[0 for _ in range(M)] for _ in range(N)]
    dist[y][x] = 1
    q.append((y, x))
    max_dist = -1
    dy = [1, -1, 0, 0]
    dx = [0, 0, 1, -1]
    while q:
        cy, cx = q.popleft()

        if dist[cy][cx] > max_dist:
            max_dist = dist[cy][cx]
        for d in range(4):
            ny = cy + dy[d]
            nx = cx + dx[d]
            if check_range(ny, nx) and board[ny][nx] == 'L':
                if dist[ny][nx] == 0:
                    dist[ny][nx] = dist[cy][cx] + 1
                    q.append((ny, nx))
    return max_dist - 1

N, M = map(int, input().split())
board = [list(map(str, input().strip())) for _ in range(N)]

answer = 0
for y in range(N):
    for x in range(M):
        if board[y][x] == 'L':
            answer = max(answer, bfs(y, x))

print(answer)

'알고리즘 풀이 > 백준' 카테고리의 다른 글

[백준 2573] 빙산  (0) 2021.06.23
[백준 2146] 다리 만들기  (0) 2021.06.21
[백준 13908] 비밀번호  (0) 2021.06.20
[백준 1342] 행운의 문자열  (0) 2021.06.18
[백준 1941] 소문난 칠공주  (0) 2021.06.18