watter bomb kotlin

dbom.kt

import kotlin.math.ceil

infix fun String.watterBomb(size: Int) = this.split("Y").map{
    ceil(it.length / size.toDouble()).toInt()
}.sum()

fun main() {
    println("xxYxx" watterBomb 3 == 2)
    println("xxYxx" watterBomb 1 == 4)
    println("xxxxYxYx" watterBomb 5 == 3)
    println("xxxxxYxYx" watterBomb 2 == 5)
    println("xxxxYxYx" watterBomb 4 == 3)
    println("xxYxx" watterBomb 3 == 2)
}