結果

問題 No.179 塗り分け
ユーザー IJKTanabeIJKTanabe
提出日時 2019-03-24 20:02:11
言語 Kotlin
(1.9.23)
結果
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
このコードへのチャレンジ
(要ログイン)

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 310 ms
57,084 KB
testcase_01 AC 315 ms
57,212 KB
testcase_02 AC 315 ms
57,032 KB
testcase_03 AC 317 ms
57,044 KB
testcase_04 AC 313 ms
57,124 KB
testcase_05 AC 487 ms
75,340 KB
testcase_06 AC 444 ms
66,828 KB
testcase_07 AC 497 ms
88,956 KB
testcase_08 AC 323 ms
57,176 KB
testcase_09 AC 340 ms
57,480 KB
testcase_10 AC 603 ms
94,468 KB
testcase_11 AC 579 ms
94,436 KB
testcase_12 AC 890 ms
93,652 KB
testcase_13 AC 329 ms
57,348 KB
testcase_14 AC 324 ms
57,200 KB
testcase_15 AC 307 ms
57,120 KB
testcase_16 AC 312 ms
57,004 KB
testcase_17 AC 320 ms
57,232 KB
testcase_18 AC 567 ms
92,368 KB
testcase_19 AC 563 ms
92,268 KB
testcase_20 AC 690 ms
92,816 KB
testcase_21 AC 654 ms
94,044 KB
testcase_22 AC 652 ms
92,924 KB
testcase_23 AC 527 ms
91,436 KB
testcase_24 AC 666 ms
92,428 KB
testcase_25 AC 567 ms
86,496 KB
testcase_26 AC 528 ms
93,316 KB
testcase_27 AC 665 ms
95,532 KB
testcase_28 AC 558 ms
94,352 KB
testcase_29 AC 668 ms
96,676 KB
testcase_30 AC 525 ms
94,300 KB
testcase_31 AC 674 ms
96,624 KB
testcase_32 AC 540 ms
94,412 KB
testcase_33 AC 688 ms
97,204 KB
testcase_34 AC 552 ms
94,500 KB
testcase_35 AC 706 ms
96,752 KB
testcase_36 AC 310 ms
57,004 KB
testcase_37 AC 321 ms
57,048 KB
testcase_38 AC 310 ms
57,092 KB
testcase_39 AC 310 ms
57,116 KB
testcase_40 AC 317 ms
57,068 KB
testcase_41 AC 311 ms
57,084 KB
testcase_42 AC 313 ms
57,132 KB
testcase_43 AC 449 ms
66,504 KB
testcase_44 AC 480 ms
77,612 KB
testcase_45 AC 336 ms
57,224 KB
権限があれば一括ダウンロードができます
コンパイルメッセージ
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