문제 설명에 앞서 아주 오랜 기간동안 포스팅을 안하다가 다시 시작한 이유는요 .... 그냥 기록하고 싶다는 생각이 다시금 들어서입니다 ㅎㅎ 그동안 논 건 아니구 꾸준히 문제 풀었고, 깃허브에 업로드 중이었습니다! 다시 열심히 하겠습니다!
문제
보물섬 지도를 발견한 후크 선장은 보물을 찾아나섰다. 보물섬 지도는 아래 그림과 같이 직사각형 모양이며 여러 칸으로 나뉘어져 있다. 각 칸은 육지(L)나 바다(W)로 표시되어 있다. 이 지도에서 이동은 상하좌우로 이웃한 육지로만 가능하며, 한 칸 이동하는데 한 시간이 걸린다. 보물은 서로 간에 최단 거리로 이동하는데 있어 가장 긴 시간이 걸리는 육지 두 곳에 나뉘어 묻혀있다. 육지를 나타내는 두 곳 사이를 최단 거리로 이동하려면 같은 곳을 두 번 이상 지나가거나, 멀리 돌아가서는 안 된다.
예를 들어 위와 같이 지도가 주어졌다면 보물은 아래 표시된 두 곳에 묻혀 있게 되고, 이 둘 사이의 최단 거리로 이동하는 시간은 8시간이 된다.
보물 지도가 주어질 때, 보물이 묻혀 있는 두 곳 간의 최단 거리로 이동하는 시간을 구하는 프로그램을 작성하시오.
풀이 방법
일단 간단한 BFS 로직을 사용하였습니다. 그래프를 전체 탐색하면서 현재 위치가 '땅'일 때 BFS 탐색을 하는 식으로 코드를 구현하였습니다. 또한 최단거리 중 최대값을 구해야 하므로 각 루프마다 최대 time(코드에선 cnt)을 갱신해 주는 식으로 했습니다.
다만, 약간 아쉬웠던 부분은 최단거리의 최대값이 가장 큰 해가 두 개 이상 존재할 때 그 중, 어떤 조건에 의해서 답이 나오도록 했으면 좋겠다고 생각을 해보았습니다. 감사합니다.
코드
def bfs(sx,sy):
max_cnt = 0
dx = [1,-1,0,0]
dy = [0,0,1,-1]
d = deque()
d.append([0,sx,sy])
visited = [[0]*M for _ in range(N)]
visited[sx][sy] = 1
while d:
cnt,x,y = d.popleft()
max_cnt = max(max_cnt,cnt)
for i in range(4):
a,b = x+dx[i],y+dy[i]
if 0<=a<N and 0<=b<M and not visited[a][b] and arr[a][b] == 'L':
visited[a][b] = 1
d.append([cnt+1,a,b])
return max_cnt
from collections import deque
import sys
input = sys.stdin.readline
N,M = map(int,input().split())
arr = [list(input().strip()) for _ in range(N)]
answer_list = []
answer = 0
for i in range(N):
for j in range(M):
if arr[i][j] == 'L':
c = bfs(i,j)
answer = max(answer,c)
print(answer)
