[코딩 테스트 연습(파이썬/Python)] 백준(BOJ) 17144번 _ 미세먼지 안녕!

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

17144번: 미세먼지 안녕!

[문제]

미세먼지를 제거하기 위해 구사과는 공기청정기를 설치하려고 한다. 공기청정기의 성능을 테스트하기 위해 구사과는 집을 크기가 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초가 지난 후 구사과의 방에 남아있는 미세먼지의 양을 구해보자.

 

[입력]

첫째 줄에 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) 주어진 미세먼지 상태를 g에 담으며 공기청정기의 위치를 찾는다.

2) 각 셀에 대해 미세먼지 확산을 시킨다.

3) 공기 청정기를 기준으로 먼지를 이동시킨다.

4) t번 반복한다.

 

# Pypy3로 제출해야 통과
import copy
r,c,t = map(int,input().split())
g = []
for num in range(r):
    tmp = list(map(int,input().split()))
    g.append(tmp)
    if tmp[0]==-1:
        cc=num
        
def dust(r,c):
    dx=(-1,1,0,0);dy=(0,0,-1,1)
    gg=copy.deepcopy(g)
    for i in range(r):
        for j in range(c):
            for k in range(4):
                if (0<=i+dx[k]<r and 0<=j+dy[k]<c) and (t:=g[i][j])!=-1:
                    if g[i+dx[k]][j+dy[k]]!=-1:
                        gg[i+dx[k]][j+dy[k]] += (t//5)
                        gg[i][j]-=(t//5)
    return gg

def push(r,c,g,cc):
    gg=copy.deepcopy(g)
    for i in range(r):
        if i ==0:
            gg[i][0:c-1] = g[i][1:]
            gg[i][-1] = g[i+1][-1]
        elif 0<i<cc-1:
            gg[i][0] = g[i-1][0]
            gg[i][-1] = g[i+1][-1]
        elif (i == cc-1) or (i==cc):
            gg[i][1:] = [0]+g[i][1:-1]
        elif cc<i<r-1:
            gg[i][0] = g[i+1][0]
            gg[i][-1] = g[i-1][-1]
        elif i ==r-1:
            gg[i][0:c-1] = g[i][1:]
            gg[i][-1] = g[i-1][-1]
    return gg
            

for _ in range(t):
    g = push(r,c,dust(r,c),cc)

print(sum(map(sum,g))+2)

파이썬은 역시나 시간초과가 난다. 어쩔 수 없나,,