[백준] 7576: 토마토

JIN·2021년 12월 18일
0

BFS

문제 풀이 전략
일반적으로 bfs 문제를 풀 때 좌표를 넣었다면 이 문제는 파라미터를 따로 넣지 않는 것이 특징입니다.
1이 여러개일때 동시에 세는 것을 어떻게 할까 고민했는데 그냥 순서대로 빼면서 bfs 돌리면 되는 것이었습니다.
그리고 기존 값에서 하나씩 더할 때 중복 되는 값이 있을 수 있으므로 혹시 모르니 방문처리는 해줍니다

from collections import deque
m, n = map(int, input().split())
graph = []
for i in range(n):
	graph.append(list(map(int, input().split())))

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

def bfs():
	while q:
		x, y = 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 graph[nx][ny] == -1:
				continue
			if graph[nx][ny] == 0:
            	graph[nx][ny] == -1
				q.append((nx, ny))
				graph[nx][ny] = graph[x][y] + 1
	return graph


bfs()
answer = 0
for i in graph:
	for j in i:
		if j == 0:
			print(-1)
			exit()
		else:
			answer = max(answer, j)
print(answer-1)
profile
배우고 적용하고 개선하기

0개의 댓글