[baekjoon]python #2636 치즈

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

2636번: 치즈

from collections import deque
col,row = map(int,input().split())
arr=[list(map(int,input().split())) for _ in range(col)]

def bfs(y,x):
    q = deque()
    q.append((y,x))
    visit=[[0]*row for _ in range(col)]
    visit[y][x]=1

    while q:
        ny,nx= q.popleft()
        directy = [-1,1,0,0]
        directx = [0,0,-1,1]

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

            if 0<=dy<col and 0<=dx<row:
                if visit[dy][dx] == 0:
                	#치즈가 있다면...
                    if arr[dy][dx]==1:
                        visit[dy][dx]=1
                        #치즈 녹이기!
                        arr[dy][dx]=0
                        continue
                    #치즈가 없다면...
                    if arr[dy][dx]==0:
                        visit[dy][dx]=1
                        q.append((dy,dx))

#치즈의 개수를 세어보자!
def count_cheeses():
    cnt_cheese=0
    for i in range(col):
        for j in range(row):
            if arr[i][j]==1:
                cnt_cheese+=1
    return cnt_cheese


time=0 #치즈가 다 녹는 시간
last_cheeses=0 #마지막 치즈

while True:
	#치즈의 개수를 세러 가기!
    cnt = count_cheeses()
    #치즈가 있다면!
    if cnt>0:
        bfs(0,0) #bfs시작!
        last_cheeses=cnt #치즈의 갯수 갱신
        time+=1 #치즈 녹는 시간 +1
        
    #치즈가 없다면!
    elif cnt==0:
    	#while문 멈춰!
        break

#출력하기
print(time)
print(last_cheeses)