알고리즘/백준(BOJ)

[백준/알고리즘] #7576: 토마토 [파이썬(python)/DFS/BFS]

https://www.acmicpc.net/problem/7576

 

7576번: 토마토

첫 줄에는 상자의 크기를 나타내는 두 정수 M,N이 주어진다. M은 상자의 가로 칸의 수, N은 상자의 세로 칸의 수를 나타낸다. 단, 2 ≤ M,N ≤ 1,000 이다. 둘째 줄부터는 하나의 상자에 저장된 토마토

www.acmicpc.net

이 문제는 시작 지점이 2곳 이상인 좌표를 동시에 BFS 구현하는 문제다.

기존에 풀었던 BFS 유형과 다르게 재귀가 아니라 큐에 토마토 판을 백업해둔 후,

방향좌표로 색칠하고 개수를 세는 문제다. 

 

 

파이썬 코드

#7576_토마토
import sys
from collections import deque
input = sys.stdin.readline

def bfs():
    dx = [-1,1,0,0]
    dy = [0,0,-1,1]

    
    while q:
        x,y = q.popleft()

        for i in range(4):
            nx = x + dx[i]
            ny = y + dy[i]

            if 0<= nx < N and 0 <= ny < M:
                if graph[nx][ny]==0:
                    q.append([nx,ny])
                    graph[nx][ny]=graph[x][y]+1


        

q=deque([])
M, N = map(int, input().split())
graph = [list(map(int, input().split())) for _ in range(N)]
path = []
for i in range(N):
    for j in range(M):
        if graph[i][j]==1:
            q.append([i,j])
            
bfs()
res=0
for i in graph:
    for j in i:
        if j==0:
            print(-1)
            exit(0)
    res = max(res, max(i))

print(res-1)