[백준 17086번] 아기 상어2 - 파이썬(python)

17086번: 아기 상어 2

분류

• 그래프 이론
• 브루트포스 알고리즘
• 그래프 탐색
• 너비 우선 탐색

문제

N×M 크기의 공간에 아기 상어 여러 마리가 있다. 공간은 1×1 크기의 정사각형 칸으로 나누어져 있다. 한 칸에는 아기 상어가 최대 1마리 존재한다.

어떤 칸의 안전 거리는 그 칸과 가장 거리가 가까운 아기 상어와의 거리이다. 두 칸의 거리는 하나의 칸에서 다른 칸으로 가기 위해서 지나야 하는 칸의 수이고, 이동은 인접한 8방향(대각선 포함)이 가능하다.

안전 거리가 가장 큰 칸을 구해보자.

입력

첫째 줄에 공간의 크기 N과 M(2 ≤ N, M ≤ 50)이 주어진다. 둘째 줄부터 N개의 줄에 공간의 상태가 주어지며, 0은 빈 칸, 1은 아기 상어가 있는 칸이다. 빈 칸과 상어의 수가 각각 한 개 이상인 입력만 주어진다.

출력

첫째 줄에 안전 거리의 최댓값을 출력한다.

풀이

• 먼저 상어의 좌표를 입력 받는다.
• BFS를 실행하면서 방문하지 않은 곳이 존재한다면 현재 좌표 +1하여 저장한다.

풀이

from collections import deque

directy = [-1, -1, -1, 0, 0, 1, 1, 1]
directx = [-1, 0, 1, -1, 1, -1, 0, 1]

n, m = map(int, input().split())
arr = [list(map(int, input().split())) for _ in range(n)]

def bfs():
    while q:
        y, x = q.popleft()
        for k in range(8):
            dy, dx = y + directy[k], x + directx[k]
            if 0 <= dy < n and 0 <= dx < m and arr[dy][dx] == 0:
                q.append((dy, dx))
                arr[dy][dx] = arr[y][x] + 1

# 상어 위치 저장하기
q = deque()
for i in range(n):
    for j in range(m):
        if arr[i][j] == 1:
            q.append((i, j))

bfs()

# 최대 거리 찾기
rlt = 0
for i in arr:
    rlt = max(rlt, max(i))

print(rlt - 1)