結果

問題 No.167 N^M mod 10
ユーザー バカらっく
提出日時 2019-11-29 05:53:52
言語 Kotlin
(2.1.0)
結果
AC  
実行時間 356 ms / 1,000 ms
コード長 1,609 bytes
コンパイル時間 12,242 ms
コンパイル使用メモリ 446,316 KB
実行使用メモリ 56,952 KB
最終ジャッジ日時 2024-11-20 21:15:27
合計ジャッジ時間 21,989 ms
ジャッジサーバーID
（参考情報） 		judge3 / judge4
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 2
other AC * 27
権限があれば一括ダウンロードができます
コンパイルメッセージ
Main.kt:4:10: warning: parameter 'arr' is never used
fun main(arr:Array<String>) {
         ^

ソースコード

diff # 

import java.util.*
import kotlin.contracts.contract

fun main(arr:Array<String>) {
    val (n,m) = (1..2).map { readLine()!! }
    val nn = n.last().toString().toInt()

    val dic = mutableMapOf<Int, Int>()

    var subTotal = 1
    var prevIdx = -1
    var toIdx = -1
    for(i in (1 until  20)) {
        subTotal *= nn
        subTotal = subTotal % 10
        if(m.length <= 3 && i == m.toInt()) {
            println(subTotal)
            return
        }
        if(m.length > 3 || m.toInt() >= 20) {
            if(dic.containsKey(subTotal)) {
                prevIdx = dic[subTotal]!!
                toIdx = i
            } else {
                dic[subTotal] = i
            }
        }
    }
    val ans = getAns(m, nn, prevIdx, toIdx)
    println(ans)
}

fun getAns(m:String, nn:Int, prevIdx:Int, toIdx:Int): Int {
    val num = m.map { it.toString().toInt() }.toMutableList()

    var subNum = prevIdx
    for(i in num.indices.reversed()) {
        if(num[i] >= subNum) {
            num[i] -= subNum
            break
        } else {
            num[i] = num[i] - subNum + 10
            subNum = 1
        }
    }
    
    val divNum = toIdx - prevIdx
    while (num.size > 1) {
        if(num[0] > divNum) {
            num[0] = num[0] % divNum
            if(num[0] == 0) {
                num.removeAt(0)
            }
        } else {
            num[1] = (num[0] * 10 + num[1]) % divNum
            num.removeAt(0)
        }
    }
    
    val cnt = prevIdx + num[0]
    
    var ans = 1
    for(i in (1..cnt)) {
        ans = ans * nn
        ans = ans % 10
    }
    return ans
    
}
0