[백준] 2636_치즈 python

김동완·2022년 4월 30일
0

알고리즘

목록 보기
33/55
post-thumbnail

치즈

문제

아래 <그림 1>과 같이 정사각형 칸들로 이루어진 사각형 모양의 판이 있고, 그 위에 얇은 치즈(회색으로 표시된 부분)가 놓여 있다. 판의 가장자리(<그림 1>에서 네모 칸에 X친 부분)에는 치즈가 놓여 있지 않으며 치즈에는 하나 이상의 구멍이 있을 수 있다.

이 치즈를 공기 중에 놓으면 녹게 되는데 공기와 접촉된 칸은 한 시간이 지나면 녹아 없어진다. 치즈의 구멍 속에는 공기가 없지만 구멍을 둘러싼 치즈가 녹아서 구멍이 열리면 구멍 속으로 공기가 들어가게 된다. <그림 1>의 경우, 치즈의 구멍을 둘러싼 치즈는 녹지 않고 ‘c’로 표시된 부분만 한 시간 후에 녹아 없어져서 <그림 2>와 같이 된다.

<그림 1> 원래 치즈 모양

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

<그림 2> 한 시간 후의 치즈 모양

<그림 3> 두 시간 후의 치즈 모양

<그림 3>은 원래 치즈의 두 시간 후 모양을 나타내고 있으며, 남은 조각들은 한 시간이 더 지나면 모두 녹아 없어진다. 그러므로 처음 치즈가 모두 녹아 없어지는 데는 세 시간이 걸린다. <그림 3>과 같이 치즈가 녹는 과정에서 여러 조각으로 나누어 질 수도 있다.

입력으로 사각형 모양의 판의 크기와 한 조각의 치즈가 판 위에 주어졌을 때, 공기 중에서 치즈가 모두 녹아 없어지는 데 걸리는 시간과 모두 녹기 한 시간 전에 남아있는 치즈조각이 놓여 있는 칸의 개수를 구하는 프로그램을 작성하시오.

입력

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

출력

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

해결방안

  • 치즈를 제거하기 위해 bfs를 사용했다.
  • 일단 가장자리에는 치즈가 없는게 보장되어 있으므로 가장자리를 큐에 넣어서 가장자리부터 탐색하게 한다.(방문처리도 한다.)
  • 치즈가 있는 부분은 remain에 더해줘서 치즈 수를 계산한다.
  • 공기에 닿은 치즈를 담을 rm 배열을 만들어준다.
  • 치즈가 있으면 while문을 도는데,
  • 처음에는 공기에 닿은 치즈가 없기 때문에 가장자리를 큐로 탐색한다.
  • 큐를 돈다.
    • rm배열에 치즈가 있다면 해당 배열을 큐로 사용한다.
    • 순회를 진행하며 그냥 공기면 큐에 넣어준다.
    • 공기에 닿은 치즈를 0으로 변경하면서 rm배열에 넣어준다. 또한 tmp에 공기에 닿은 치즈들을 더해준다.
    • 방문처리한다.
  • 큐가 끝나면 remain에서 tmp를 빼준다.
  • rm배열에 값이 있으면 해당 배열을 큐로 사용해서 while문을 돈다.
  • remain이 0이면 끝낸다
from collections import deque 
n,m = map(int,input().split())

#그리드 받아오기 
grid = [list(map(int,input().split())) for _ in range(n)]
#방향 설정 
dr = [0,-1,0,1]
dc = [1,0,-1,0]

#큐 만들기 
q = deque()
#남은 치즈는 remain
remain = 0
#방문배열 
visited =[]

#배열을 돌면서 
for r in range(n) :
    for c in range(m) :
        #가장자리를 큐에 넣는다. 그리고 방문처리한다. 
        if r == 0 or c == 0 or r == n-1 or c == m-1 :
            q.append([r,c])
            visited.append((r,c))
        #치즈가 잇으면 remain에 추가해준다. 
        elif grid[r][c] == 1 :
            remain += 1
#공기에 담은 치즈를 담을 배열 
rm = []
#반복횟수 
cnt = 0
while True :
    cnt +=1
    #공기에 닿은 치즈들이 담겨있으면 치즈를 큐로 사용  
    if rm :
        tmp_q = deque(rm[:])
    #아니면 가장자리를 큐로 사용 
    else :
        tmp_q = q
    rm = []
    
    tmp = 0
    #큐에 값이 있으면 
    while tmp_q :
        # 큐에서 원소를 꺼내서 
        cr,cc = tmp_q.popleft()
        # 4방향을 순회한다. 
        for d in range(4) :
            nr = cr + dr[d]
            nc = cc + dc[d] 
            #순회한 방향이 범위 안에 잇으면서 방문하지 않았을 때 
            if 0<nr<n-1 and 0<nc<m-1 and not (nr,nc) in visited :
                #1이면 공기에 닿았다고 판단한다. 
                #공기에 닿은 부분만 제거할것이기 때문에 큐에는 넣지 않는다.
                if grid[nr][nc] == 1 :
                    rm.append([nr,nc])
                    #공기에 닿으면 공기가 되므로 0으로 바꾼다. 
                    grid[nr][nc] = 0
                    #공기에 닿은 치즈 수 
                    tmp +=1 
                #그냥 공기면 큐에 넣어줘서 계속 탐색하게 한다. 
                elif grid[nr][nc] == 0 :
                    tmp_q.append([nr,nc])
                #다시 방문하지 않도록 방문처리한다. 
                visited.append((nr,nc)) 
    #남은 치즈에서 공기에 닿은 치즈를 빼준다. 
    remain = remain-tmp
    #남은 치즈가 없으면 종료한다. 
    if remain == 0 :
        break

#반복횟수 
print(cnt)
#마지막 남았던 치즈 수 
print(tmp)
profile
내가 공부한 내용들이 누군가에게 도움이 될지 몰라서 쓰는 벨로그

0개의 댓글