package yukicoder.no24 /** * エントリポイント * @param args コマンドライン引数 */ fun main(args: Array) { val in1 = readLine() val in2 = mutableListOf() var line: String? line = readLine() while (line != null) { in2.add(line) line = readLine() } println(numbersGame(in1, in2)) } /** * 数あてゲームの回答を返します。 * @param turn ターン数 * @param qaList 質問と回答 */ fun numbersGame(@Suppress("UNUSED_PARAMETER") turn: String?, qaList: List): String { // int型の配列を10個作る。初期値は0 // 0:初期値、1:NO、2:YES val numAry = IntArray(10, { 0 }) for (qa in qaList) { val sp = qa.split(" ") val qList = sp.take(sp.size - 1).map { it.toInt() } val a = sp[sp.size - 1] if (a == "NO") { for (q in qList) { numAry[q] = 1 } } if (a == "YES") { // 0と2以外はすべて1 for ((i, num) in numAry.withIndex()) { if ((num == 0 || num == 2) && qList.any { it == i }) { numAry[i] = 2 } else { numAry[i] = 1 } } } } // 結果を探す for ((i, num) in numAry.withIndex()) { if (num == 2) { return i.toString() } } for ((i, num) in numAry.withIndex()) { if (num == 0) { return i.toString() } } // ここには来ないはず return "" }