[BOJ][🟡5][백준#02589] 보물섬

문제 출처

BAEKJOON Online Judge #2589

문제

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

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

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

입력

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

출력

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

예제

입력

5 7
WLLWWWL
LLLWLLL
LWLWLWW
LWLWLLL
WLLWLWW

출력

8

My Sol

import sys
input = sys.stdin.readline

def BFS(i, j):
    visited = [[0]*J for _ in range(I)]
    didj = [(-1,0),(1,0),(0,1),(0,-1)]
    visited[i][j] = 1
    Q = [(i, j)]
    front = -1
    rear = 0
    max_now = 0
    while front < rear:
        front += 1
        ni, nj = Q[front]
        for di, dj in didj:
            si, sj = ni+di, nj+dj
            if not (0<=si<I and 0<=sj<J): continue
            if visited[si][sj]: continue
            if not mat[si][sj]: continue
            visited[si][sj] = visited[ni][nj]+1
            max_now = max(visited[si][sj], max_now)
            Q.append((si, sj))
            rear += 1
    return max_now-1


I, J = map(int, input().split())
mat = []
for _ in range(I):
    l = [1 if i=='L' else 0 for i in list(input())]
    mat.append(l)

max_val = 0
for i in range(I):
    for j in range(J):
        if not mat[i][j]: continue
        val = BFS(i, j)
        if max_val < val:
            max_val = val
print(max_val)

BFS를 이용하는 문제였다. 완전탐색을 이용해 전체 칸에 대해 조회하는데 만약 ‘L’ 칸을 만나면 상하좌우의 가장 인접한 칸마다 시간을 더해가며 채우는 것이다. 그러다가 채울 수 있는 칸의 마지막 칸의 값을 매번 최댓값과 비교하며 완전탐색을 마치면 해당 최댓값이 지도에서 가장 갈 수 있는 제일 먼 곳까지의 소요 시간이 된다.

결과

맞았습니다!!

모범답안

출처

import sys, collections
n, m = [int(x) for x in sys.stdin.readline().split()]
board = [list(sys.stdin.readline().rstrip()) for _ in range(n)]
dx, dy = (1, -1, 0, 0), (0, 0, 1, -1)
def bfs(x, y):
    q = collections.deque([(x, y)])
    visited = [[0]*m for _ in range(n)]
    visited[x][y] = 1
    mx = 0
    while q:
        x, y = q.popleft()
        for i in range(4):
            nx, ny = x+dx[i], y+dy[i]
            if not (0 <= nx < n and 0 <= ny < m) or visited[nx][ny] != 0 or board[nx][ny] == "W": continue
            visited[nx][ny] = visited[x][y] + 1
            mx = max(visited[nx][ny], mx)
            q.append((nx, ny))
    return mx-1
result = 0
for i in range(n):
    for j in range(m):
        if board[i][j] == "L":
            if 0 <= i-1 and i+1 < n:
                if board[i-1][j] == 'L' and board[i+1][j] == 'L': continue
            if 0 <= j-1 and j+1 < m:
                if board[i][j-1] == 'L' and board[i][j+1] == 'L': continue
            result = max(result, bfs(i, j))
print(result)

연산 시간이 짧은 답이 있어서 가져와 보았다. 완전탐색을 하는 경우를 가지치기해주었는데, 사방이 육지인 곳부터는 절대 최대길이가 나올 수 없는 점을 이용해서 bfs를 시작할 때 상하좌우 모두 ‘L’ 있다면 제외하고 시작하는 것이다. 똑똑한 방법이라고 생각한다.