// No.5 数字のブロック // https://yukicoder.me/problems/no/5 package no01.yukicoder.no5 /** * インプットデータ */ data class InputData( /** * 箱の幅 */ val boxWidth: Int, /** * ブロックの数 */ val blockNum: Int, /** * 各ブロックの幅 */ val blockWidth: List ) /** * エントリポイント */ fun main(args: Array) { val input = getStandardInput() println(block(input)) } /** * 箱に入るブロックの数を計算して返します。 */ fun block(input: List): Int { val data = createInputData(input) // sorted で List をソート。 // sortedDescending()で降順にソート val sortedBlock = data.blockWidth.sorted() var inBlockNum = 0 var inBlockWidth = 0 // 小さいブロックから数えて、箱の幅を超えたら終了 for (v in sortedBlock) { if (v + inBlockWidth <= data.boxWidth) { inBlockNum++ inBlockWidth += v } else { break } } return inBlockNum } /** * 標準入力から取得した文字列をInputDataに変換して返します。 */ fun createInputData(input: List): InputData { val blockWidth = input[2] .split(" ") .map { it.toInt() } return InputData(input[0].toInt(), input[1].toInt(), blockWidth) } /** * 標準入力から文字列を全て取得します。 */ fun getStandardInput(): List { val lines = mutableListOf() var line: String? // readLine で標準入力から文字列を取得 line = readLine() // 入力がなくなるまで取得する while (line != null) { lines.add(line) line = readLine() } return lines }