🐹 미세먼지 안녕!

1. 문제 링크

17144번: 미세먼지 안녕!

2. 코드

Python3 31256KB 4372ms

PyPy3 165244KB 728ms

R, C, T = map(int, input().split())
room = []
air_cleaner = []
for i in range(R):
    line = list(map(int, input().split()))
    if -1 in line:
        air_cleaner.append(i)
    room.append(line)

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

# 미세먼지 확산
def diffusion(x, y):
    count = 0
    move_amount = room[x][y] // 5
    for i in range(4):
        nx = x + dx[i]
        ny = y + dy[i]
        # 벽에 안 부딪히고 공기 청정기 아닐 시 확산
        if 0 <= nx < R and 0 <= ny < C and room[nx][ny] != -1:
            count += 1
            dust[nx][ny][1] += move_amount  # 확산
    dust[x][y][0] = move_amount * count # 확산되고 줄은 양

# 공기청정기 작동
def air_clean():
    # 반시계 방향 
    for i in range(air_cleaner[0], 0, -1):
        room[i][0], room[i-1][0] = room[i-1][0], room[i][0]
    for i in range(0, C-1):
        room[0][i], room[0][i+1] = room[0][i+1], room[0][i]
    for i in range(0, air_cleaner[0]):
        room[i][C-1], room[i+1][C-1] = room[i+1][C-1], room[i][C-1]
    for i in range(C-1, 0, -1):
        room[air_cleaner[0]][i], room[air_cleaner[0]][i-1] = room[air_cleaner[0]][i-1], room[air_cleaner[0]][i]
    room[air_cleaner[0]][1] = 0     # 깨끗한 바람
    
    # 시계 방향
    for i in range(air_cleaner[1], R-1):
        room[i][0], room[i+1][0] = room[i+1][0], room[i][0]
    for i in range(0, C-1):
        room[R-1][i], room[R-1][i+1] = room[R-1][i+1], room[R-1][i]
    for i in range(R-1, air_cleaner[1], -1):
        room[i][C-1], room[i-1][C-1] =  room[i-1][C-1], room[i][C-1]
    for i in range(C-1, 0, -1):
        room[air_cleaner[1]][i], room[air_cleaner[1]][i-1] = room[air_cleaner[1]][i-1], room[air_cleaner[1]][i]
    room[air_cleaner[1]][1] = 0     # 깨끗한 바람

while T:
    T -= 1
    # 줄어들 먼지 양, 늘어날 먼지 양
    dust = [[[0, 0] for _ in range(C)] for _ in range(R)]
    # 미세먼지 이동 확인
    for i in range(R):
        for j in range(C):
            if room[i][j] > 0:
                diffusion(i, j)
    # 미세먼지 확산 정리
    for i in range(R):
        for j in range(C):
            # 변동사항이 있을 때
            if dust[i][j] != [0, 0]:
                room[i][j] -= dust[i][j][0]
                room[i][j] += dust[i][j][1]
    air_clean()

# 남아있는 미세먼지 양
result = 0
for i in range(R):
    for j in range(C):
        if room[i][j] > 0:
            result += room[i][j]
print(result)

3. 해설

시뮬레이션을 사용하여 문제를 해결했다.

1. 미세먼지가 확산될 때 모든 방의 미세먼지가 확산되기 전에 값이 변경되면 최종 값이 다르게 나오므로 dust 3차원 배열에 각 위치에 줄어들 먼지양, 늘어날 먼지양 값을 담아두었다.

2. 공기 청정기가 돌 때 도는 방향과 반대 방향으로 이동하면서 값을 스위칭 해주었다. 한 바퀴를 돌았으면 공기청정기의 바람 시작 위치의 값만 0으로 변경해준다. (공기 청정기에서 나오는 바람은 깨끗한 바람이므로)

위의 과정을 T 시간동안 반복해준다.