[Python] 백준 17144번 미세먼지 안녕!

골드 5

문제

미세먼지를 제거하기 위해 구사과는 공기청정기를 설치하려고 한다. 공기청정기의 성능을 테스트하기 위해 구사과는 집을 크기가 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. 공기청정기가 작동한다.
   • 공기청정기에서는 바람이 나온다.
   • 위쪽 공기청정기의 바람은 반시계방향으로 순환하고, 아래쪽 공기청정기의 바람은 시계방향으로 순환한다.
   • 바람이 불면 미세먼지가 바람의 방향대로 모두 한 칸씩 이동한다.
   • 공기청정기에서 부는 바람은 미세먼지가 없는 바람이고, 공기청정기로 들어간 미세먼지는 모두 정화된다.

다음은 확산의 예시이다.

입력

첫째 줄에 R, C, T (6 ≤ R, C ≤ 50, 1 ≤ T ≤ 1,000) 가 주어진다.

둘째 줄부터 R개의 줄에 Ar,c (-1 ≤ Ar,c ≤ 1,000)가 주어진다. 공기청정기가 설치된 곳은 Ar,c가 -1이고, 나머지 값은 미세먼지의 양이다. -1은 2번 위아래로 붙어져 있고, 가장 윗 행, 아랫 행과 두 칸이상 떨어져 있다.

출력

첫째 줄에 T초가 지난 후 구사과 방에 남아있는 미세먼지의 양을 출력한다.

 

---

시뮬레이션

1. 미세먼지의 확산

--> 변화값은 바로 처리하지 않고 별도의 배열을 만들어 저장 후 한번에 더해주어야 함.

--> 진행 과정에 영향 줄 수 있기 때문

2. 공기 청정기 작동

--> 순환 방향에 따른 2가지 함수 필요

--> 각각 dx, dy 진행 방향에 맞게 정의

 

 

# 백준 17144번 미세먼지 안녕!

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

def spread():
    dx = [1, -1, 0, 0]
    dy = [0, 0, 1, -1]  
    change = [[0] * c for _ in range(r)] # amount of dust that is diffused
    for x in range(r):
        for y in range(c):
            if graph[x][y] > 0: # if dust exists
                cnt = 0
                for i in range(4): # for 4 directions
                    nx = x + dx[i]
                    ny = y + dy[i]
                    if 0 <= nx < r and 0 <= ny < c and graph[nx][ny] != -1:
                        change[nx][ny] += graph[x][y] // 5
                        cnt += 1
                graph[x][y] -= graph[x][y] // 5 * cnt
    for x in range(r):
        for y in range(c):
            graph[x][y] += change[x][y] # add diffused dust to the graph
    return

def top_rotate():
    dx = [0, -1, 0, 1] # set direction of the top rotation
    dy = [1, 0, -1, 0]
    d = 0
    before = 0
    x, y = top_cleaner, 1 # start from right slot of the top_cleaner
    while True:
        nx = x + dx[d]
        ny = y + dy[d]
        if x == top_cleaner and y == 0: # Back to cleaner position -> terminate
            break
        if nx < 0 or nx >= r or ny < 0 or ny >= c: # if out of the map
            d += 1 # change direction and continue
            continue
        graph[x][y], before = before, graph[x][y] # swap the slots
        x, y = nx, ny # move to next slot
    return


def down_rotate():
    dx = [0, 1, 0, -1] # set direction of the bottom rotation
    dy = [1, 0, -1, 0]
    d = 0
    before = 0
    x, y = bottom_cleaner, 1
    while True:
        nx = x + dx[d]
        ny = y + dy[d]
        if x == bottom_cleaner and y == 0:
            break
        if nx < 0 or nx >= r or ny < 0 or ny >= c:
            d += 1
            continue
        graph[x][y], before = before, graph[x][y]
        x, y = nx, ny
    return

top_cleaner, bottom_cleaner = 0, 0

for i in range(r): # find position of cleaner
    if graph[i][0] == -1:
        top_cleaner = i
        bottom_cleaner = i + 1
        break

for _ in range(t): # for t seconds
    spread()
    top_rotate()
    down_rotate()

answer = 2 # because 2 slots of cleaner are -1
for i in range(r):
    answer += sum(graph[i])
print(answer)

---

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

17144번: 미세먼지 안녕!