[백준 / Python] 7576번 토마토 | 초코더

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

7576번: 토마토

 

접근법

다른 부분은 앞의 게시글과 거의 동일합니다. 참고해주세요!

https://sinsomi.tistory.com/entry/%EB%B0%B1%EC%A4%80-Python-2178%EB%B2%88-%EB%AF%B8%EB%A1%9C-%ED%83%90%EC%83%89-%EC%B4%88%EC%BD%94%EB%8D%94?category=854286

[백준 / Python] 2178번 미로 탐색 | 초코더

다른점은 앞의 문제는 queue에 첫 시작점을 넣고 다음좌표로 상하좌우 비교해 주었다면

            토마토가 1인 좌표를 queue에다가 저장한후, 그 queue를 차례로 비교해주는것입니다.

 

나의 풀이

import sys
from collections import deque

m,n=list(map(int,sys.stdin.readline().split()))
tomato=[list(map(int,sys.stdin.readline().split())) for _ in range(n)]
visited = [[-1]*m for _ in range(n)]

dx = [-1,0,1,0]
dy = [0,1,0,-1]
queue=deque() 
for i in range(n):
    for j in range(m):
        if tomato[i][j]==1:
            visited[i][j]=0 
            queue.append([i,j])

while queue:
    x,y=queue.popleft() 
    for i in range(4):
        nx=x+dx[i]
        ny=y+dy[i]
        if nx<0 or nx>=n or ny<0 or ny>=m:
            continue
        if tomato[nx][ny] == 0 and visited[nx][ny] == -1:
            queue.append([nx,ny])
            visited[nx][ny]=visited[x][y]+1

ans=max([max(row) for row in visited]) #이해안감
for i in range(n):
    for j in range(m):
        if tomato[i][j]==0 and visited[i][j]==-1:
            ans=-1
print(ans)