문제
보물섬 지도를 발견한 후크 선장은 보물을 찾아나섰다. 보물섬 지도는 아래 그림과 같이 직사각형 모양이며 여러 칸으로 나뉘어져 있다. 각 칸은 육지(L)나 바다(W)로 표시되어 있다. 이 지도에서 이동은 상하좌우로 이웃한 육지로만 가능하며, 한 칸 이동하는데 한 시간이 걸린다. 보물은 서로 간에 최단 거리로 이동하는데 있어 가장 긴 시간이 걸리는 육지 두 곳에 나뉘어 묻혀있다. 육지를 나타내는 두 곳 사이를 최단 거리로 이동하려면 같은 곳을 두 번 이상 지나가거나, 멀리 돌아가서는 안 된다.
예를 들어 위와 같이 지도가 주어졌다면 보물은 아래 표시된 두 곳에 묻혀 있게 되고, 이 둘 사이의 최단 거리로 이동하는 시간은 8시간이 된다.
보물 지도가 주어질 때, 보물이 묻혀 있는 두 곳 간의 최단 거리로 이동하는 시간을 구하는 프로그램을 작성하시오.
입력
첫째 줄에는 보물 지도의 세로의 크기와 가로의 크기가 빈칸을 사이에 두고 주어진다. 이어 L과 W로 표시된 보물 지도가 아래의 예와 같이 주어지며, 각 문자 사이에는 빈 칸이 없다. 보물 지도의 가로, 세로의 크기는 각각 50이하이다.
출력
첫째 줄에 보물이 묻혀 있는 두 곳 사이를 최단 거리로 이동하는 시간을 출력한다.
예제 입력5 7WLLWWWL LLLWLLL LWLWLWW LWLWLLL WLLWLWW |
예제 출력8 |
풀이
L인 부분을 모두 탐색하며 BFS를 진행했을 때, 가장 긴 최단 거리를 출력하면 된다.
기본적으로 브루트 포스 + BFS로 진행한다.
여기서 브루트 포스만 활용한다면 PyPy3 로만 AC를 받는 것이 가능하고, 백트래킹까지 한다면 Python3로도 AC가 가능하다.
백트래킹의 핵심은 L인 부분을 탐색할 때, 상하 혹은 좌우에 L이 동시에 존재한다면 그곳은 가장자리가 아니기 때문에 탐색을 넘기는 것이다.
from collections import deque
import sys
input = sys.stdin.readline
n,m = map(int,input().split())
board = [list(input().rstrip()) for _ in range(n)]
visit = [[0]*m for _ in range(n)]
move = [(0,1),(1,0),(0,-1),(-1,0)]
ans = 0
for i in range(n):
for j in range(m):
# 가장자리가 아님
if i>0 and i+1<n:
if board[i-1][j] == "L" and board[i+1][j] == "L":
continue
if j>0 and j+1<m:
if board[i][j-1] == "L" and board[i][j+1] == "L":
continue
# BFS 진행
if board[i][j] == "L" and not visit[i][j]:
q = deque([(i,j)])
visit_tmp = [i[:] for i in visit]
visit_tmp[i][j] = 1
r = 0
while q:
x,y = q.popleft()
for a,b in move:
dx=x+a; dy=y+b
if n>dx>=0 and m>dy>=0 and not visit_tmp[dx][dy] and board[dx][dy]=="L":
visit_tmp[dx][dy] = visit_tmp[x][y] + 1
r = max(r,visit_tmp[dx][dy])
q.append((dx,dy))
ans = max(ans,r)
print(ans-1)
'알고리즘 연습 > DFS와 BFS' 카테고리의 다른 글
[🥇3 / 백준 1937 / 파이썬] 욕심쟁이 판다 (0) | 2021.10.16 |
---|---|
[🥇5 / 백준 2636 / 파이썬] 치즈 (0) | 2021.10.15 |
[🥇4 / 백준 2573 / 파이썬] 빙산 (0) | 2021.10.14 |
[🥇2 / 백준 16946 / 파이썬] 벽 부수고 이동하기 4 (0) | 2021.10.07 |
[🥈1 / 백준 1389 / 파이썬] 케빈 베이컨의 6단계 법칙 (0) | 2021.10.07 |