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) { 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>() for (move in validMove) { when (move) { "S" -> { val state = gameState.copy() state.nextTurnS() nextSteps.add(Pair("S", state)) } "R" -> { val state = gameState.copy() state.nextTurnR() nextSteps.add(Pair("R", state)) } "L" -> { val state = gameState.copy() state.nextTurnL() nextSteps.add(Pair("L", state)) } } } val minState = nextSteps.sortedWith( compareBy( { pair -> pair.second.enemyList.count { it.isAlive() } }, { it.second.evaluateScore() }) ).first() gameState = minState.second println(minState.first) } } class GameState { var enemyList = mutableListOf() 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 { 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.up() val first = enemyList.filter { it.isAlive() && it.x == own.x && it.y > own.y }.sortedBy { it.y }.firstOrNull() ?: return first.hit(own.level()) if (!first.isAlive()) { own.defeat(first.p) } } fun nextTurnR() { turn++ own.up() own.right() val first = enemyList.filter { it.isAlive() && it.x == own.x && it.y > own.y }.sortedBy { it.y }.firstOrNull() ?: return first.hit(own.level()) if (!first.isAlive()) { own.defeat(first.p) } } fun nextTurnL() { turn++ own.up() own.left() val first = enemyList.filter { it.isAlive() && it.x == own.x && it.y > own.y }.sortedBy { it.y }.firstOrNull() ?: return 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.copy() gameState.turn = turn return gameState } fun evaluateScore(): Int { val collision = enemyList.filter { it.x == own.x && (it.y == own.y || it.y == own.y + 1) } val first = enemyList.filter { it.isAlive() && it.x == own.x && it.y > own.y }.minBy { it.y } if (collision.isNotEmpty()) { return Int.MAX_VALUE } if (first != null && first.h > (first.y - own.y) * own.level()) { return Int.MAX_VALUE } return enemyList.filter { it.isAlive() }.sumBy { it.h } } } 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 } fun up() { y++ } fun right() { x++ } fun left() { x-- } fun copy(): Own { val copy = Own(x, y) copy.p = p return copy } }