QA & Engineering Blog

A Blog about Quality · Automation · Engineering

🏠 홈으로

17144. 미세먼지 안녕!

업데이트 시간 : 2023-03-13 14:00:20 +0000

[Gold IV] 미세먼지 안녕! - 17144

문제 링크

성능 요약

메모리: 181476 KB, 시간: 3324 ms

분류

구현, 시뮬레이션

문제 설명

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

💡 Solutions

📄 미세먼지 안녕!.py

R,C,T = map(int,input().split())

rooms = [list(map(int,input().split())) for _ in range(R)]
dx = (-1,0,1,0)
dy = (0,1,0,-1)

def flying_munzi(x,y, _munzi : dict):
    mz = rooms[x][y]
    if not _munzi.get((x,y)):
        _munzi[(x,y)] = 0
    diffuse = mz//5
    for k in range(4):
        nx = x + dx[k]
        ny = y + dy[k]
        if nx<0 or ny<0 or nx>=R or ny>=C:
            continue
        if rooms[nx][ny] == -1:
            continue

        if not _munzi.get((nx,ny)):
            _munzi[(nx,ny)] = 0

        _munzi[(nx,ny)] += diffuse
        mz -= diffuse
    _munzi[(x,y)] += mz
    return _munzi

def clean_munzi(clnr):

    top_x, top_y = clnr[0]
    bot_x, bot_y = clnr[1]

    top_moving = 0
    top_x = top_x + dx[top_moving]
    top_y = top_y + dy[top_moving]
    next_tx = top_x
    next_ty = top_y

    while rooms[top_x][top_y] != -1:
        
        if next_tx <= 0 or next_tx >= clnr[0][0] or next_ty <= 0 or next_ty >= C - 1:
            if (next_tx == 0 and next_ty == 0) or (next_tx == 0 and next_ty == C - 1) or (next_tx == clnr[0][0] and next_ty == C - 1):
                top_moving = (top_moving +1)%4

        next_tx = top_x + dx[top_moving]
        next_ty = top_y + dy[top_moving]

        if rooms[next_tx][next_ty] != -1:
            rooms[top_x][top_y] = rooms[next_tx][next_ty]
            rooms[next_tx][next_ty] = 0

        top_x = next_tx
        top_y = next_ty



    bot_moving = 2
    bot_x = bot_x + dx[bot_moving]
    bot_y = bot_y + dy[bot_moving]

    next_bx = bot_x
    next_by = bot_y
    while rooms[bot_x][bot_y] != -1:
        if next_bx <= clnr[1][0] or next_bx >= R-1 or next_by <= 0 or next_by >= C - 1:
            if (next_bx == R-1 and next_by == 0) or (next_bx == R-1 and next_by == C - 1) or (next_bx == clnr[1][0] and next_by == C - 1):
                bot_moving = (bot_moving -1)%4

        next_bx = bot_x + dx[bot_moving]
        next_by = bot_y + dy[bot_moving]

        if rooms[next_bx][next_by] != -1:
            rooms[bot_x][bot_y] = rooms[next_bx][next_by]
            rooms[next_bx][next_by] = 0

        bot_x = next_bx
        bot_y = next_by
    

def time_out():
    time = 0

    while time <T:
        time += 1

        munzi = dict()
        cleaner =[]
        for i in range(R):
            for j in range(C):
                if rooms[i][j] == -1 :
                    cleaner.append((i,j))
                elif rooms[i][j] != 0:
                    munzi = flying_munzi(i,j, munzi)
                    rooms[i][j] = 0
        
        for a,b in munzi:
            rooms[a][b] = munzi[(a,b)]

        clean_munzi(cleaner)

    rlt = 0
    for i in range(R):
        for j in range(C):
            if rooms[i][j] != -1 and rooms[i][j] != 0:
                rlt += rooms[i][j]

    return rlt

print(time_out())