문제

미세먼지를 제거하기 위해 구사과는 공기청정기를 설치하려고 한다. 공기청정기의 성능을 테스트하기 위해 구사과는 집을 크기가 R×C인 격자판으로 나타냈고, 1×1 크기의 칸으로 나눴다. 구사과는 뛰어난 코딩 실력을 이용해 각 칸 (r, c)에 있는 미세먼지의 양을 실시간으로 모니터링하는 시스템을 개발했다. (r, c)는 r행 c열을 의미한다.

공기청정기는 항상 1번 열에 설치되어 있고, 크기는 두 행을 차지한다. 공기청정기가 설치되어 있지 않은 칸에는 미세먼지가 있고, (r, c)에 있는 미세먼지의 양은 Ar,c이다.

1초 동안 아래 적힌 일이 순서대로 일어난다.

1. 미세먼지가 확산된다. 확산은 미세먼지가 있는 모든 칸에서 동시에 일어난다.
   • (r, c)에 있는 미세먼지는 인접한 네 방향으로 확산된다.
   • 인접한 방향에 공기청정기가 있거나, 칸이 없으면 그 방향으로는 확산이 일어나지 않는다.
   • 확산되는 양은 Ar,c/5이고 소수점은 버린다.
   • (r, c)에 남은 미세먼지의 양은 Ar,c - (Ar,c/5)×(확산된 방향의 개수) 이다.
2. 공기청정기가 작동한다.
   • 공기청정기에서는 바람이 나온다.
   • 위쪽 공기청정기의 바람은 반시계방향으로 순환하고, 아래쪽 공기청정기의 바람은 시계방향으로 순환한다.
   • 바람이 불면 미세먼지가 바람의 방향대로 모두 한 칸씩 이동한다.
   • 공기청정기에서 부는 바람은 미세먼지가 없는 바람이고, 공기청정기로 들어간 미세먼지는 모두 정화된다.

방의 정보가 주어졌을 때, T초가 지난 후 구사과의 방에 남아있는 미세먼지의 양을 구해보자.

# 17144
import sys
input = lambda: sys.stdin.readline().strip()

# 1. 미세먼지 확산
# 1-1. 미세먼지는 인접한 네 방향으로 확산
# 1-2. 인접 방향에 공기청정기가 있거나, 칸이 없으면 확산 X
# 1-3. 확산되는 양은 A[r][c] // 5
# 1-4. A[r][c] = A[r][c] - A[r][c] // 5 * (확산된 방향의 개수)

# 2. 공기청정기 작동
# 2-1. 위쪽 공기청정기의 바람은 반시계방향으로 순환
# 2-2. 아래쪽 공기청정기의 바람은 시계방향으로 순환
# 2-3. 바람이 불면 미세먼지가 바람의 방향대로 한 칸씩 이동
# 2-4. 공기청정기로 미세먼지가 들어가면 정화됨

from collections import deque

r, c, t = map(int, input().split())
arr = [list(map(int, input().split())) for _ in range(r)]

dx, dy = [-1, 0, 1, 0], [0, 1, 0, -1]
ax1, ax2 = -1, -1


def pm(count_pm):
    global r, c, arr
    queue = deque(count_pm)
    
    add_arr = [[0] * c for _ in range(r)]
    
    while queue:
        now_pm = queue.popleft()
        x, y = now_pm[0], now_pm[1]
        
        cnt = 0
        for i in range(4):
            nx, ny = x + dx[i], y + dy[i]
            if nx < 0 or ny < 0 or nx >= r or ny >= c or arr[nx][ny] == -1:
                continue
            
            cnt += 1
            add_arr[nx][ny] += arr[x][y] // 5
                
        arr[x][y] -= arr[x][y] // 5 * cnt
    
    arr = [[arr[i][j] + add_arr[i][j] for j in range(c)] for i in range(r)]
            

def air(ax1, ax2):
    global r, c
    for i in range(ax1-1, -1, -1):
        arr[i+1][0], arr[i][0] = arr[i][0], 0
    for i in range(ax2+1, r):
        arr[i-1][0], arr[i][0] = arr[i][0], 0
        
    for j in range(1, c):
        arr[0][j-1], arr[0][j] = arr[0][j], 0
        arr[r-1][j-1], arr[r-1][j] = arr[r-1][j], 0
        
    for i in range(1, ax1+1):
        arr[i-1][c-1], arr[i][c-1] = arr[i][c-1], 0
    for i in range(r-2, ax2-1, -1):
        arr[i+1][c-1], arr[i][c-1] = arr[i][c-1], 0
        
    for j in range(c-2, 0 ,-1):
        arr[ax1][j+1], arr[ax1][j] = arr[ax1][j], 0
        arr[ax2][j+1], arr[ax2][j] = arr[ax2][j], 0
    
    arr[ax1][0], arr[ax2][0] = -1, -1


def cnt_pm(count_pm):
    global r, c, ax1, ax2
    for i in range(r):
        for j in range(c):
            if arr[i][j] == -1:
                if ax1 == -1:
                    ax1 = i
                else:
                    ax2 = i
            elif arr[i][j]:
                count_pm.append([i, j])
    
    
for _ in range(t):
    count_pm = []
    cnt_pm(count_pm)
    
    pm(count_pm)
    
    air(ax1, ax2)

total = 2
for a in arr:
    total += sum(a)
print(total)

백준 17144 미세먼지 안녕!