結果

問題 No.103 素因数ゲーム リターンズ
コンテスト
ユーザー バカらっく
提出日時 2019-09-30 09:07:44
言語 Kotlin
(2.1.0)
結果
WA  
実行時間 -
コード長 2,112 bytes
コンパイル時間 11,282 ms
コンパイル使用メモリ 449,244 KB
実行使用メモリ 118,356 KB
最終ジャッジ日時 2024-10-03 05:08:35
合計ジャッジ時間 28,097 ms
ジャッジサーバーID
（参考情報） 		judge5 / judge2
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 4 WA * 1
other AC * 5 WA * 2 TLE * 1 -- * 12
権限があれば一括ダウンロードができます
コンパイルメッセージ
Main.kt:3:10: warning: parameter 'arr' is never used
fun main(arr:Array<String>) {
         ^
Main.kt:4:9: warning: variable 'itemCount' is never used
    val itemCount = readLine()!!.toInt()
        ^

ソースコード

diff # 

import kotlin.math.max

fun main(arr:Array<String>) {
    val itemCount = readLine()!!.toInt()
    val numberList = readLine()!!.split(" ").map { convDic[it.toInt()] }.sorted()
    val ans = if(canWin(numberList.sorted())) "Alice" else "Bob"
    println(ans)
}

val dic = mutableMapOf<String, Boolean>()
fun canWin(numberList:List<Int>):Boolean {
    if(numberList.isEmpty() || numberList.all { it == 1 }) {
        return false
    }
    val key = numberList.joinToString("-")
    dic[key]?.let { return it }

    val maxIndex = numberList.lastIndex
    var canWin = false
    for(i in (0..maxIndex)) {
        if(numberList[i] == 1) {
            continue
        }
        val prime = getPrimeDivide(numberList[i]).toMutableMap()
        for(p in prime.keys) {
            for(j in (1..Math.min(2,prime[p]!!))) {
                val dv = Math.pow(p.toDouble(), j.toDouble()).toInt()
                val newList = numberList.toMutableList()
                newList[i] = numberList[i] / dv
                if(!canWin(newList.sorted())) {
                    canWin = true
                    break
                }
            }
            if(canWin) {
                break
            }
        }
        if(canWin) {
            break
        }
    }
    dic[key] = canWin
    return canWin
}

val primeDic = (0..10000).map { getPrimeDivide(it) }
val convDic = (0..10000).map { getNewValue(primeDic[it]) }
fun getNewValue(prime: Map<Int,Int>):Int {
    val list = prime.map { Math.pow(it.key.toDouble(), it.value.toDouble()).toInt() }
    if(prime.isEmpty()) {
        return 1
    }
    val ans = list.reduce { acc, i -> acc * i }
    return ans
}
fun getPrimeDivide(num:Int):Map<Int, Int> {
    var div = 1
    var current = num
    var ret = mutableMapOf<Int, Int>()
    while (current > 1) {
        div++
        var cnt = 0
        while (current % div == 0) {
            cnt++
            current /= div
        }
        if(cnt > 0) {
            ret[div] = cnt % 3
        }
    }
    ret = (2 until 2 + ret.count()).map { it to ret.values.sorted()[it-2]}.toMap().toMutableMap()
    return ret
}
0