// No.21 平均の差 // https://yukicoder.me/problems/no/21 package no01.yukicoder.no21 /** * インプットデータ */ data class InputData( /** * 数字の数 */ val numCount: Int, /** * グループ */ val group: Int, /** * 数字 */ val nums: List ) /** * エントリポイント */ fun main(args: Array) { val input = getStandardInput() println(average(input)) } fun average(input: List): Int { val data = createInputData(input) val sortList = data.nums.sorted() return sortList.last() - sortList.first() } /** * 標準入力から取得した文字列をInputDataに変換して返します。 */ fun createInputData(input: List): InputData { val numCount = input[0].toInt() val group = input[1].toInt() val nums = mutableListOf() for ((i, v) in input.withIndex()) { if (i == 0 || i == 1) { continue } nums.add(v.toInt()) } return InputData(numCount, group, nums) } /** * 標準入力から文字列を全て取得します。 */ fun getStandardInput(): List { val lines = mutableListOf() var line: String? line = readLine() while (line != null) { lines.add(line) line = readLine() } return lines }