結果

問題 No.179 塗り分け
ユーザー IJKTanabe
提出日時 2019-03-24 20:02:11
言語 Kotlin
(2.1.0)
結果
AC  
実行時間 890 ms / 3,000 ms
コード長 2,784 bytes
コンパイル時間 15,662 ms
コンパイル使用メモリ 448,144 KB
実行使用メモリ 97,204 KB
最終ジャッジ日時 2024-07-23 14:59:44
合計ジャッジ時間 39,066 ms
ジャッジサーバーID
（参考情報） 		judge4 / judge1
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 6
other AC * 40
権限があれば一括ダウンロードができます
コンパイルメッセージ
Main.kt:1:10: warning: parameter 'args' is never used
fun main(args: Array<String>) {
         ^
Main.kt:28:5: warning: parameter 'cellsList' is never used
    cellsList: List<List<String>>,
    ^
Main.kt:31:5: warning: parameter 'W' is never used
    W: Int,
    ^

ソースコード

diff # 

fun main(args: Array<String>) {
    val (H, W) = readLine()!!.split(" ").map(String::toInt)
    // 入力を各セルごとの二次元リストに
    val cellsList: List<List<String>> = List(H) { readLine()!!.chunked(1) }
    // 行ごとに、黒色セルのインデックスのリストを取得
    val blackIndices = cellsList.map {
        it.mapIndexedNotNull { index, value ->
            if (value == "#") index else null
        }
    }

/*
    //Check Print
    cellsList.forEach { println(it) }
    blackIndices.forEach { println(it) }
*/

    for (vecY in (-H + 1)..(H - 1)) for (vecX in (-W + 1)..(W - 1)) {
        if (tryPaintB(cellsList, blackIndices, H, W, vecY, vecX)) {
            println("YES")
            return
        }
    }
    println("NO")
}

fun tryPaintB(
    cellsList: List<List<String>>,
    blackIndices: List<List<Int>>,
    H: Int,
    W: Int,
    vecY: Int,
    vecX: Int
): Boolean {

    //val cells = cellsList.map { it.toMutableList() }
    val rangeY =
        if (vecY < 0)
            ((H - 1) downTo 0).toList()
        else
            (0 until H).toList()

    val indices =
        if (vecX < 0)
            blackIndices.map { it.reversed().toMutableList() }
        else
            blackIndices.map { it.toMutableList() }

    var flg: Boolean? = null

/*
    //Check Print
    println("vec = $vecY, $vecX")
    println("Y_range = $rangeY")
    println("blackIndices =")
    indices.forEach { println(it) }
*/

    for (y in rangeY) {
        while (indices[y].isNotEmpty()) {
            var trgY = y
            var trgX = indices[y].first()

            while (rangeY.contains(trgY) && indices[trgY].contains(trgX)) {
                //cells[y][x] = "R"
                indices[trgY].remove(trgX)
                trgY += vecY
                trgX += vecX
                if (rangeY.contains(trgY) && indices[trgY].contains(trgX)) {
                    //cells[trgY][trgX] = "B"
                    indices[trgY].remove(trgX)
                    if (flg == null) flg = true
                    trgY += vecY
                    trgX += vecX
                } else {
                    //flg = false
                    return false
                }
            }
//            if (rangeY.contains(trgY) && indices[trgY].contains(trgX)) {
//                //cells[trgY][trgX] = "B"
//                indices[trgY].remove(trgX)
//                if (flg == null) flg = true
//            } else {
//                //flg = false
//                return false
//            }
        }
    }

/*
    //Check Print
    cells.forEach { println(it.joinToString("")) }
    println(flg)
    println(" ")
*/

    return flg ?: false
}
0