🟩 [백준] [Python] [Gold4] 2636번_치즈

문제

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

2636번: 치즈

풀이

# 백준 2636번 문제 - 치즈
from collections import deque
dx, dy = [0,0,-1,1], [1,-1,0,0]
n, m = map(int, input().split())

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

def bfs():
    visited = [[0]*m for _ in range(n)]
    
    q = deque()
    q.append((0,0))
    visited[0][0] = 1
    cnt = 0
    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 and not visited[nx][ny]:
                if graph[nx][ny] == 0:
                    visited[nx][ny] = 1
                    q.append((nx, ny))
                elif graph[nx][ny] == 1:
                    graph[nx][ny] = 0
                    cnt += 1
                    visited[nx][ny] = 1
    cheese.append(cnt)
    return cnt

cheese = []        
time = 0
while True:
    time += 1
    cnt = bfs()
    
    if cnt == 0:
        break
print(time-1)
print(cheese[-2])

1을 찾아서 BFS를 하는 것이 아닌 0에서 BFS를 실행해야 한다. 

거꾸로 생각하면 쉬운 문제였는데 문제를 잘못 이해해서 오래 걸렸다.