[백준 2636번] 파이썬 - 치즈

백준 2636 - 치즈

시간 제한 1초, 메모리 제한 128MB

# 조건

• 아래 <그림 1>과 같이 정사각형 칸들로 이루어진 사각형 모양의 판이 있고, 그 위에 얇은 치즈(회색으로 표시된 부분)가 놓여 있다.
• 판의 가장자리(<그림 1>에서 네모 칸에 X친 부분)에는 치즈가 놓여 있지 않으며 치즈에는 하나 이상의 구멍이 있을 수 있다.
• 이 치즈를 공기 중에 놓으면 녹게 되는데 공기와 접촉된 칸은 한 시간이 지나면 녹아 없어진다.
• 치즈의 구멍 속에는 공기가 없지만 구멍을 둘러싼 치즈가 녹아서 구멍이 열리면 구멍 속으로 공기가 들어가게 된다.
• <그림 1>의 경우, 치즈의 구멍을 둘러싼 치즈는 녹지 않고 ‘c’로 표시된 부분만 한 시간 후에 녹아 없어져서 <그림 2>와 같이 된다.

• 다시 한 시간 후에는 <그림 2>에서 ‘c’로 표시된 부분이 녹아 없어져서 <그림 3>과 같이 된다.

• <그림 3>은 원래 치즈의 두 시간 후 모양을 나타내고 있으며, 남은 조각들은 한 시간이 더 지나면 모두 녹아 없어진다.
• 그러므로 처음 치즈가 모두 녹아 없어지는 데는 세 시간이 걸린다.
• <그림 3>과 같이 치즈가 녹는 과정에서 여러 조각으로 나누어 질 수도 있다.
• 입력으로 사각형 모양의 판의 크기와 한 조각의 치즈가 판 위에 주어졌을 때, 공기 중에서 치즈가 모두 녹아 없어지는 데 걸리는 시간과 모두 녹기 한 시간 전에 남아있는 치즈조각이 놓여 있는 칸의 개수를 구하는 프로그램을 작성하시오.

입력

• 첫째 줄에는 사각형 모양 판의 세로와 가로의 길이가 양의 정수로 주어진다.
• 세로와 가로의 길이는 최대 100이다.
• 판의 각 가로줄의 모양이 윗 줄부터 차례로 둘째 줄부터 마지막 줄까지 주어진다.
• 치즈가 없는 칸은 0, 치즈가 있는 칸은 1로 주어지며 각 숫자 사이에는 빈칸이 하나씩 있다.

출력

• 첫째 줄에는 치즈가 모두 녹아서 없어지는 데 걸리는 시간을 출력하고, 둘째 줄에는 모두 녹기 한 시간 전에 남아있는 치즈조각이 놓여 있는 칸의 개수를 출력한다.

# 접근 방법

• bfs를 이용하여 풀어주면 된다.
• 다만 이 때, 치즈를 기준으로 돌려주는 것이 아닌 외부 공기를 기준으로 돌려주어야 한다.
• 입력받은 배열을 순회하며 외부 공기를 접촉하기 위하여 탐색을 시작하며 치즈의 개수도 기록해준다.
  • 가장 바깥 테두리는 무조건 외부 공기이다.
  • 따라서, 0, 0에서 bfs를 시작하여 준다.
• 외부 공기를 찾다가 치즈를 만났다면 temp에 기록해주고 외부 공기를 모두 찾은 후 치즈를 공기로 변경해준다.
  • 최초에 기록된 치즈의 개수에서 cnt 만큼 빼주며 기록해준다.
• 만약 모든 치즈가 녹았다면 시간과, 직전의 cnt를 출력한다.

import sys  
sys.stdin = open('input.txt')  
input = sys.stdin.readline  
from collections import deque  

def bfs(ci, cj):  
    q = deque()  
    q.append((ci, cj))  
    visited = [[0]*M for _ in range(N)]  
    visited[ci][cj] = 1  
    while q:  
        si, sj = q.popleft()  
        for d in range(4):  
            ni, nj = si+di[d], sj+dj[d]  
            if 0<=ni<N and 0<=nj<M and visited[ni][nj] == 0:  
                if arr[ni][nj] == 0:  
                    q.append((ni, nj))  
                    visited[ni][nj] = 1  
                elif arr[ni][nj] == 1:  
                    temp.append((ni, nj))  
                    visited[ni][nj] = 1  

    for ti, tj in temp:  
        arr[ti][tj] = 0  

    return len(temp)  

N, M = map(int, input().split())  
arr = [[*map(int, input().split())] for _ in range(N)]  
di, dj = [1, -1, 0, 0], [0, 0, 1, -1]  

cheese_cnt = 0  
melt_cnt = 0  

for i in range(N):  
    for j in range(M):  
        if arr[i][j] == 1:  
            cheese_cnt += 1  

while True:  
    temp = []  
    melt_cnt += 1  
    val = bfs(0, 0)  
    if cheese_cnt - val == 0:  
        print(melt_cnt)  
        print(cheese_cnt)  
        break  
    else:  
        cheese_cnt -= val