結果

問題 No.5017 Tool-assisted Shooting
コンテスト
ユーザー エセ賢者
提出日時 2023-08-03 13:28:39
言語 Kotlin
(2.1.0)
結果
RE  
実行時間 -
コード長 3,821 bytes
コンパイル時間 18,972 ms
コンパイル使用メモリ 440,624 KB
実行使用メモリ 71,692 KB
スコア 68,472
平均クエリ数 1.03
最終ジャッジ日時 2023-08-03 13:29:39
合計ジャッジ時間 57,756 ms
ジャッジサーバーID
（参考情報） 		judge13 / judge12
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
other RE * 100
権限があれば一括ダウンロードができます
コンパイルメッセージ
Main.kt:10:10: warning: parameter 'args' is never used
fun main(args: Array<String>) {
         ^
Main.kt:46:59: warning: unnecessary non-null assertion (!!) on a non-null receiver of type GameState
        gameState = nextSteps.minBy { it.evaluateScore() }!!
                                                          ^
Main.kt:118:23: warning: 'sumBy((T) -> Int): Int' is deprecated. Use sumOf instead.
            enemyList.sumBy { it.h }
                      ^

ソースコード

diff #
raw source code 

private fun next() = readLine()!!
private fun nextInt() = next().toInt()
private fun nextLong() = next().toLong()
private fun nextDouble() = next().toDouble()
private fun nextList() = next().split(" ")
private fun nextIntList() = next().split(" ").map { it.toInt() }.toIntArray()
private fun nextLongList() = next().split(" ").map { it.toLong() }.toLongArray()
private fun nextDoubleList() = next().split(" ").map { it.toDouble() }.toDoubleArray()

fun main(args: Array<String>) {
    var gameState = GameState()

    while (!gameState.isDone()) {
        val n = nextInt()
        if (n == -1) {
            return
        }
        for (i in 0 until n) {
            val hpx = nextIntList()
            gameState.addEnemy(hpx)
        }

        val validMove = gameState.validMove()
        val nextSteps = mutableListOf<GameState>()
        for (move in validMove) {
            when (move) {
                "S" -> {
                    val state = gameState.copy()
                    state.nextTurnS()
                    nextSteps.add(state)
                }

                "R" -> {
                    val state = gameState.copy()
                    state.nextTurnR()
                    nextSteps.add(state)
                }

                "L" -> {
                    val state = gameState.copy()
                    state.nextTurnL()
                    nextSteps.add(state)
                }
            }
        }
        gameState = nextSteps.minBy { it.evaluateScore() }!!
    }
}

class GameState {
    var enemyList = mutableListOf<Enemy>()
    var own = Own(12, 0)
    var turn = 0

    fun addEnemy(hpx: IntArray) {
        enemyList.add(Enemy(hpx[2], 59 + turn, hpx[0], hpx[1]))
    }

    fun isDone(): Boolean {
        return turn >= 1000
    }

    fun validMove(): Array<String> {
        val move = mutableListOf("S")
        if (enemyList.none { it.x == own.x + 1 && it.y == own.y }) {
            move.add("R")
        }
        if (enemyList.none { it.x == own.x - 1 && it.y == own.y }) {
            move.add("L")
        }
        return move.toTypedArray()
    }

    fun nextTurnS() {
        turn++
        own = Own(own.x, own.y + 1)
        println("S")
        val first = enemyList.first { it.isAlive() && it.x == own.x && it.y > own.y }
        first.hit(own.level())
        if (!first.isAlive()) {
            own.defeat(first.p)
        }
    }

    fun nextTurnR() {
        turn++
        own = Own(own.x + 1, own.y + 1)
        println("R")
        val first = enemyList.first { it.isAlive() && it.x == own.x && it.y > own.y }
        first.hit(own.level())
        if (!first.isAlive()) {
            own.defeat(first.p)
        }
    }

    fun nextTurnL() {
        turn++
        own = Own(own.x - 1, own.y + 1)
        println("L")
        val first = enemyList.first { it.isAlive() && it.x == own.x && it.y > own.y }
        first.hit(own.level())
        if (!first.isAlive()) {
            own.defeat(first.p)
        }
    }

    fun copy(): GameState {
        val gameState = GameState()
        gameState.enemyList = enemyList.map { Enemy(it.x, it.y, it.h, it.p) }.toMutableList()
        gameState.own = own
        gameState.turn = turn
        return gameState
    }

    fun evaluateScore(): Int {
        val collision = enemyList.filter { it.x == own.x && it.y == own.y }
        return if (collision.isEmpty()) {
            enemyList.sumBy { it.h }
        } else {
            Int.MAX_VALUE
        }
    }
}

class Enemy(var x: Int, var y: Int, var h: Int, var p: Int) {
    fun isAlive(): Boolean {
        return h > 0
    }

    fun hit(damage: Int) {
        h -= damage
    }
}

class Own(var x: Int, var y: Int) {
    var p = 100
    fun defeat(power: Int) {
        p += power
    }

    fun level(): Int {
        return p / 100
    }
}
0