[백준][코틀린] 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. 공기청정기가 작동한다.
   • 공기청정기에서는 바람이 나온다.
   • 위쪽 공기청정기의 바람은 반시계방향으로 순환하고, 아래쪽 공기청정기의 바람은 시계방향으로 순환한다.
   • 바람이 불면 미세먼지가 바람의 방향대로 모두 한 칸씩 이동한다.
   • 공기청정기에서 부는 바람은 미세먼지가 없는 바람이고, 공기청정기로 들어간 미세먼지는 모두 정화된다.

예시는 백준 사이트 참고 하세용!!

입력

첫째 줄에 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초가 지난 후 구사과 방에 남아있는 미세먼지의 양을 출력한다.

예제 입력	예제 출력
7 8 1 0 0 0 0 0 0 0 9 0 0 0 0 3 0 0 8 -1 0 5 0 0 0 22 0 -1 8 0 0 0 0 0 0 0 0 0 0 0 10 43 0 0 0 5 0 15 0 0 0 0 0 40 0 0 0 20 0	188

 

문제 풀이

시뮬레이션 문제는 문제 조건만 따라가면, 풀 수 있다! 시뮬레이션 문제인 만큼 코드에 조건을 달아놨다. 

이상하거나 잘 모르는 부분이 있으면 언제든 댓글 달아주세용🙆🏻‍♀️

</>̆̈ 코드

import java.util.*

// 미세먼지 안녕!

private var airCleaner = -1
private var stepR = listOf(0, 0, 1, -1)
private var stepC = listOf(1, -1, 0, 0)

fun main() {
    val br = System.`in`.bufferedReader()
    val (r, c, time) = br.readLine().split(" ").map { it.toInt() }
    var map = Array(r) {StringTokenizer(br.readLine()).run { IntArray(c) {nextToken().toInt()} }}

    repeat(time) {
        val nMap = Array(r) {IntArray(c)}

        // 미세먼지 확산
        for (row in 0 until r) {
            for (col in 0 until c) {
            	
                // 공기청정기의 위치를 저장시킨다. (한번 초기화 되었으면 다시 초기화 못시키게 조건 검)
                if (airCleaner == -1 && map[row][col] == -1)  airCleaner = row

                // 공기청정기가 없고(==-1) 먼지가 없는 곳(==0)을 제외하고 살펴본다.
                if (map[row][col] > 0) {
                	
                    var cnt = 0 // 확산 시킨 횟수
                    val dust = map[row][col] / 5 // 확산 시킬 먼지 양

                    for (step in 0 until 4) {
                        val nRow = row + stepR[step]; val nCol = col + stepC[step]

                        // 맵의 범위 안이고 그 위치에 공기청정기가 없다면 확산시킨다.
                        if (nRow in 0 until r && nCol in 0 until c && map[nRow][nCol] != -1) {
                            nMap[nRow][nCol] += dust
                            cnt++
                        }
                    }
                    // 먼지 확산시키고 줄어든 원래 칸
                    nMap[row][col] += map[row][col] - dust * cnt
                }
            }
        }
        // 공기 청정기 부분 업데이트
        nMap[airCleaner][0] = -1; nMap[airCleaner+1][0] = -1

        // 공기청정기 작동 (윗부분 문제에 주어진 순서대로 밀어줌)
        for (row in airCleaner - 1 downTo 1) nMap[row][0] = nMap[row-1][0]
        for (col in 0 until c -1) nMap[0][col] = nMap[0][col+1]
        for (row in 0 until  airCleaner) nMap[row][c-1] = nMap[row+1][c-1]
        for (col in c - 1 downTo 2) nMap[airCleaner][col] = nMap[airCleaner][col-1]
        nMap[airCleaner][1] = 0 // 공기 청정기에서 나온 공기

        // 공기청정기 작동 (아랫부분)
        for (row in airCleaner + 2 until r - 1) nMap[row][0] = nMap[row + 1][0]
        for (col in 0 until c - 1) nMap[r-1][col] = nMap[r-1][col+1]
        for (row in r-1 downTo  airCleaner+2) nMap[row][c-1] = nMap[row-1][c-1]
        for (col in c-1 downTo 2) nMap[airCleaner+1][col] = nMap[airCleaner+1][col-1]
        nMap[airCleaner+1][1] = 0 // 공기 청정기에서 나온 공기

        map = nMap
    }

    var result = 2 // 합을 구할 때 -1이 두개 있으니까(공기청정기 있는자리)
    for (i in 0 until r) result += map[i].sum()

    print(result)
}

링크

17144번: 미세먼지 안녕!