[맛있는물회] [파이썬] <백준 알고리즘> 4864번 "숫자카드"

문제 조건

---

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

 

Input

---

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

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

 

Output

---

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

 

생각한 아이디어

---

단순 시뮬레이션 문제이다.

 

문제에서 주어진 흐름대로 쭉쭉 나아가면 된다.

 

하지만 몇가지 신경써야할 부분이 있다.

먼저 미세먼지 Spread 부분이다.

단순하게 2중 포문으로 MAP을 돌면서 미세먼지 일때 확산되게 짜버리면 level이 안 맞는 경우가 발생한다. 즉, 첫번째 확산을 통해 만들어진 미세먼지는 두번째 확산에서 확산이 일어나야 하는데 반복문의 조건이 0 보다 클 때로 해서 첫번째 확산에서 확산이 일어날 수 있다.  visited 나 새로운 배열을 만들어서 copy 하는 방식으로 해결 가능하다.

 

그리고 공기청정기를 통해 미세먼지가 움직일 때 Corner에 있는 미세먼지를 움직여 주는 부분에 조금 신경써줘야 한다. 

 

그 외에는 알고리즘의 흐름만 잘 따라서 코드를 짜준다면 쉽게 해결할 수 있는 문제이다.

 

소스코드

#17144 미세먼지 안녕!
import sys
from collections import deque
import copy
input=sys.stdin.readline
inf = 2*32

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

def spread():
    air_map=[[0 for _ in range(c)] for _ in range(r)]
    for y in range(0,r):
        for x in range(0,c):
            if MAP[y][x]>0:#미세먼지
                cnt=0
                for dir in range(0,4):
                    ny=y+dy[dir]
                    nx=x+dx[dir]
                    val=MAP[y][x]//5
                    if r>ny>=0 and c>nx>=0 and MAP[ny][nx]>=0:
                        cnt+=1
                        air_map[ny][nx]+=val
                air_map[y][x] += MAP[y][x]- val*cnt
    return air_map

r,c,t=map(int,input().split())
MAP=[list(map(int,input().split())) for _ in range(r)]
air=[]
for y in range(0,r):
    for x in range(0,c):
        if MAP[y][x]==-1:air.append(y)

for _ in range(t):
    tmp = spread()
    tmp[air[0]][0],tmp[air[1]][0]=-1,-1
    MAP=copy.deepcopy(tmp)

    MAP2=[[0 for _ in range(c)] for _ in range(r)]

    a = air[0]
    for j in range(1,c-1):
        MAP2[a][j+1] = MAP[a][j]
    MAP2[a-1][c-1] = MAP[a][c-1]
    for j in range(a,0,-1):
        MAP2[j-1][c-1] = MAP[j][c-1]
    MAP2[0][c-2]=MAP[0][c-1]
    for j in range(c-1,0,-1):
        MAP2[0][j-1] = MAP[0][j]
    MAP2[1][0] = MAP[0][0]
    for j in range(0,a-1):
        MAP2[j+1][0] = MAP[j][0]
    
    a = air[1]
    for j in range(1,c-1):
        MAP2[a][j+1] = MAP[a][j]
    MAP2[a+1][c-1] = MAP[a][c-1]
    for j in range(a,r-1):
        MAP2[j+1][c-1] = MAP[j][c-1]
    MAP2[r-1][c-2]=MAP[r-1][c-1]
    for j in range(c-1,0,-1):
        MAP2[r-1][j-1] = MAP[r-1][j]
    MAP2[r-2][0] = MAP[r-1][0]
    for j in range(r-1,a-1,-1):
        MAP2[j-1][0] = MAP[j][0]
    
    MAP2[air[0]][0],MAP2[air[1]][0]=-1,-1
    for y in range(1,air[0]):
        for x in range(1,c-1):
            MAP2[y][x]=MAP[y][x]
    for y in range(air[1]+1,r-1):
        for x in range(1,c-1):
            MAP2[y][x]=MAP[y][x]
    MAP=copy.deepcopy(MAP2)

SUM=2
for row in range(0,r):
    SUM+=sum(MAP[row])
print(SUM)

 

*파이썬 문법 정리