[알고리즘] 백준 7576 토마토

CHOI IN HO·2024년 3월 5일
0

코딩테스트

목록 보기
61/74

풀이

bfs를 활용해서 풀었고, 현재 1로 되어있는 부분만 q에 넣는다. 이후 인접한 과일들을 먼저 +1씩해주고 가장 큰 숫자를 출력

from collections import deque
m, n = map(int, input().split())

array = []
for i in range(n):
    array.append(list(map(int, input().split())))

q = deque()
for i in range(n):
    for j in range(m):
        if array[i][j] == 1:
            q.append((i, j , 1))

dx = [-1, 1, 0, 0]
dy = [0, 0, -1, 1]
def bfs(q):
    count = 0
    while q:
        x, y, count = q.popleft()
        for i in range(4):
            nx = x+dx[i]
            ny = y+dy[i]
            if nx < 0 or ny < 0 or nx >= n or ny >= m:
                continue
            if array[nx][ny] == 0:
                array[nx][ny] = count
                q.append((nx,ny,count+1))

    return count
mx = 0
count = bfs(q)
ans = True
for k in array:
    mx = max(mx, max(k))
    if 0 in k:
        ans = False
        break
if ans == True:
    if count == 1:
        print(0)
    else:print(mx)
else:
    print(-1)
profile
개발자기 되기 위해선 무엇이든!

0개의 댓글