[백준] 14719번: 빗물 - Kotlin[코틀린]

문제

 

14719번: 빗물

 

---

풀이

 

 입력받은 블록 나타내는 2차원 배열을 생성하여 해결하였다. 배열에서 1은 블록을 나타내고 0은 빈 공간을 나타내도록 구성했다.

 

 2중 for문을 사용하여 1층부터 확인한다. 처음 블록을 만나면 해당 블록 다음부터 빈 공간의 개수를 세고, 다시 블록을 만나면 빈 공간에 빗물이 고일수 있는 것으로 처리하여 전체 결과값에 더해준다. 모든 층을 확인하고 결과값을 출력해주면 된다.

 

---

코드

 

fun main() {
    val br = System.`in`.bufferedReader()
    val bw = System.out.bufferedWriter()

    val (height, width) = br.readLine().split(' ').map { it.toInt() }
    val block = br.readLine().split(' ').map { it.toInt() }
    val map = Array(height){ IntArray(width) }

    for(i in 0 until width) {
        for(j in 0 until block[i]) {
            map[j][i] = 1
        }
    }
    
    var sum = 0
    for(i in 0 until height) {
        var check = false
        var cnt = 0
        for(j in 0 until width) {
            if(check) {
                if(map[i][j] == 0) cnt++
                else {
                    sum += cnt
                    cnt = 0
                }
            } else {
                if(map[i][j] == 1) check = true
            }
        }
    }

    bw.write("$sum")
    bw.flush()
    bw.close()
    br.close()
}