본문 바로가기

알고리즘/알고리즘 문제

[python] 백준 > BFS > 보물섬(2589)

[문제]

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

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

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

입력

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

출력

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

www.acmicpc.net/problem/2589

 

2589번: 보물섬

보물섬 지도를 발견한 후크 선장은 보물을 찾아나섰다. 보물섬 지도는 아래 그림과 같이 직사각형 모양이며 여러 칸으로 나뉘어져 있다. 각 칸은 육지(L)나 바다(W)로 표시되어 있다. 이 지도에서

www.acmicpc.net

[풀이]

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

n, m = map(int, input().split())
arr = [list(input().rstrip()) for _ in range(n)]
dx, dy = [-1, 1, 0, 0], [0, 0, -1, 1]
answer = []

def bfs(x, y):
    q = deque()
    q.append((x, y))
    visited = [[-1]*m for _ in range(n)]
    visited[x][y]=0
    while q:
        x, y = q.popleft()
        for i in range(4):
            sx, sy = x+dx[i], y+dy[i]
            if not (0<=sx<n and 0<=sy<m):
                continue
            if arr[sx][sy]=='L' and visited[sx][sy]==-1:
                visited[sx][sy] = visited[x][y]+1
                q.append((sx, sy))
    return max([max(val) for val in visited])


for a in range(n):
    for b in range(m):
        if arr[a][b]=='L':
            if 0<=a-1 and a+1<n:
                if arr[a-1][b]=='L' and arr[a+1][b]=='L':
                    continue
            if 0<=b-1 and b+1<m:
                if arr[a][b-1]=='L' and arr[a][b+1]=='L':
                    continue
            answer.append(bfs(a, b))

print(max(answer))
  • 모든 L에 대해서 각각의 최댓값을 구한다. 하지만 그러면 시간초과가 나기때문에 조건을 더해준다.
  • 내 현재위치 기준 양옆 또는 위아래에도 땅일 경우 더 긴시간으로 갈 수 있는 땅이 존재하는 것이기 때문에 bfs를 호출하지 않도록 해준다.
  • visited를 -1로 초기화해준뒤에, 내 현재위치 기준 동서남북으로 이동했을 때 땅이라면 현재위치 + 1을 이동한 위치의 값으로 넣어준다.