정보
문제 바로가기 [클릭]
난이도: Gold5
관련 개념: #그래프이론 #그래프탐색 #브루트포스 #너비 우선 탐색
조건
시간 제한 | 메모리 제한 |
---|---|
1 초 | 512 MB |
문제
보물섬 지도를 발견한 후크 선장은 보물을 찾아나섰다. 보물섬 지도는 아래 그림과 같이 직사각형 모양이며 여러 칸으로 나뉘어져 있다. 각 칸은 육지(L)나 바다(W)로 표시되어 있다. 이 지도에서 이동은 상하좌우로 이웃한 육지로만 가능하며, 한 칸 이동하는데 한 시간이 걸린다. 보물은 서로 간에 최단 거리로 이동하는데 있어 가장 긴 시간이 걸리는 육지 두 곳에 나뉘어 묻혀있다. 육지를 나타내는 두 곳 사이를 최단 거리로 이동하려면 같은 곳을 두 번 이상 지나가거나, 멀리 돌아가서는 안 된다.
예를 들어 위와 같이 지도가 주어졌다면 보물은 아래 표시된 두 곳에 묻혀 있게 되고, 이 둘 사이의 최단 거리로 이동하는 시간은 8시간이 된다.
보물 지도가 주어질 때, 보물이 묻혀 있는 두 곳 간의 최단 거리로 이동하는 시간을 구하는 프로그램을 작성하시오.
입력
첫째 줄에는 보물 지도의 세로의 크기와 가로의 크기가 빈칸을 사이에 두고 주어진다. 이어 L과 W로 표시된 보물 지도가 아래의 예와 같이 주어지며, 각 문자 사이에는 빈 칸이 없다. 보물 지도의 가로, 세로의 크기는 각각 50이하이다.
출력
첫째 줄에 보물이 묻혀 있는 두 곳 사이를 최단 거리로 이동하는 시간을 출력한다.
예제 입출력 1
입력
1
2
3
4
5
6
5 7
WLLWWWL
LLLWLLL
LWLWLWW
LWLWLLL
WLLWLWW
출력
1
8
코드(파이썬)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
from collections import deque
def bfs(y, x):
queue = deque([[y, x]])
visited = [[0] * m for _ in range(n)]
visited[y][x] = 1
cnt = 0
while queue:
for _ in range(len(queue)):
now_y, now_x = queue.popleft()
for i in range(4):
next_x = now_x + dx[i]
next_y = now_y + dy[i]
if 0 <= next_x < m and 0 <= next_y < n and treasure[next_y][next_x] and visited[next_y][next_x] == 0:
visited[next_y][next_x] = 1
queue.append((next_y, next_x))
cnt += 1
return cnt-1
n, m = map(int, input().split())
dx = [1, -1, 0, 0]
dy = [0, 0, 1, -1]
treasure = [[1 if ch == "L" else 0 for ch in input().rstrip()] for _ in range(n)]
shortest = 0
for i in range(n):
for j in range(m):
if treasure[i][j]:
shortest = max(shortest, bfs(i, j))
print(shortest)
특이사항
- 해결방법
- 단순한 BFS를 최적화해서 해결
- BFS 의사코드를 확인하고 그동안 잘못 사용했던 부분을 수정함
참고문헌
-
Comments powered by Disqus.