結果

問題 No.117 組み合わせの数
ユーザー 8989
提出日時 2022-02-15 07:33:00
言語 Kotlin
(1.9.23)
結果
AC  
実行時間 1,816 ms / 5,000 ms
コード長 4,575 bytes
コンパイル時間 15,983 ms
コンパイル使用メモリ 466,260 KB
実行使用メモリ 235,156 KB
最終ジャッジ日時 2024-06-29 07:02:57
合計ジャッジ時間 20,101 ms
ジャッジサーバーID
（参考情報） 		judge2 / judge3
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 1,816 ms
235,156 KB
権限があれば一括ダウンロードができます

ソースコード

diff # 

import java.io.BufferedReader
import java.io.InputStream
import java.io.InputStreamReader
import java.io.PrintWriter
import java.lang.StringBuilder
import java.util.*
const val inf:Long = 1000000000
var mod99:Long = 998244353
var mod10:Long = 1000000007
val reader = System.`in`.bufferedReader()
//val out = PrintWriter(System.out)
fun main() {
    var comb = combination(2000001)
    var t = one_To_Int()
    for (i in 0 until t) {
        var x = input()
        if (x[0] == 'C') {
            var (n,m) = x.substring(2,x.length - 1).split(",").map{ it.toInt() }
            println(comb.C(n,m))
        }
        if (x[0] == 'P') {
            var (n,m) = x.substring(2,x.length - 1).split(",").map{ it.toInt() }
            println(comb.P(n,m))
        }
        if (x[0] == 'H') {
            var (n,m) = x.substring(2,x.length - 1).split(",").map{ it.toInt() }
            println(comb.H(n,m))
        }

    }
    //out.flush()
    reader.close()
}
class combination(size:Int){
    var fac = MutableList<Long>(size) {0}
    var inv = MutableList<Long>(size) {0}
    init {
        //階乗mod p
        var now_fac = 1L
        fac[0] = 1
        inv[0] = 1
        for (i in 1..size - 1){
            now_fac *= i.toLong()
            now_fac %= mod10
            fac[i] = now_fac
        }
        //逆元
        var f = mod_pow(now_fac, mod10-2, mod10)
        inv[0] = f
        var i = size-1
        while (i != 0){
            f = f * i % mod10
            inv[size - i] = f
            i -= 1
        }
        inv = inv.asReversed()
    }
    fun C(n:Int,m:Int):Long{
        if (n == -1 && m == 0) return 1
        if (0 > m || m > n) return 0
        return ((fac[n]*inv[m])%mod10)*inv[n - m]%mod10
    }
    fun P(n:Int,m:Int):Long{
        if (0 > m || m > n) return 0
        return ((fac[n]*inv[n - m])%mod10)
    }
    fun H(n:Int,k:Int):Long{
        if (n == k && n == 0) return 1
        if ((n == 0 && k > 0) || k < 0) return 0
        return C(n + k - 1,k)
    }

}
fun init_String_SortedList():SortedSet<String>{
    return init_String_List().toSortedSet<String>()
}
fun init_Int_SortedList():SortedSet<Int>{
    return init_Int_List().toSortedSet<Int>()
}
fun init_Long_SortedList():SortedSet<Long>{
    return init_Long_List().toSortedSet<Long>()
}
fun mod_pow(x:Long, n:Long, p: Long): Long {
    var ans = 1L
    var X: Long = x
    var N: Long = n
    while (N >= 1L) {
        if (N % 2 == 1L) {
            ans *= X
            ans = ans % p
        }
        X *= X
        X = X % p
        N = N / 2
    }
    return ans
}
fun min(a:MutableList<Long>):Long{
    var now :Long = 10000000000
    for (i in 0..a.size - 1){
        if (now > a[i]){
            now = a[i]
        }
    }
    return now
}
fun max(a:MutableList<Long>):Long{
    var now :Long = -100000000000
    for (i in 0..a.size - 1){
        if (now < a[i]){
            now = a[i]
        }
    }
    return now
}
fun input():String{
    return reader.readLine()
}
fun init_Long_List():MutableList<Long>{
    var a = MutableList<Long>(0) { 0 }
    return a
}
fun init_String_List():MutableList<String>{
    var a = MutableList<String>(0) { "" }
    return a
}
fun init_Int_List():MutableList<Int>{
    var a = MutableList<Int>(0) { 0 }
    return a
}
fun list_To_Int():MutableList<Int>{
    return input().split(" ").map{it.toInt()}.toMutableList()
}
fun list_To_Long():MutableList<Long>{
    return input().split(" ").map{it.toLong()}.toMutableList()
}
fun one_To_Int():Int{
    return input().toInt()
}
fun one_To_Long():Long {
    return input().toLong()
}
//fun deque():ArrayDeque<Long>{
//    var queue:ArrayDeque<Long> = ArrayDeque()
//    return queue
//}
fun merge(A:MutableList<Long>,left:Int, mid:Int, right:Int) {
    var n1 = mid - left;
    var n2 = right - mid;
    var L = MutableList<Long>(n1 + 1) { 0 }
    var R = MutableList<Long>(n2 + 1) { 0 }
    for (i in 0..n1 - 1) {
        L[i] = A[left + i]
    }
    for (i  in 0..n2 - 1){
        R[i] = A[mid + i]
    }
    L[n1] = inf
    R[n2] = inf
    var i = 0
    var j = 0
    for (k in left..right - 1) {

        if (L[i] <= R[j]) {
            A[k] = L[i]
            i = i + 1
        }
        else {
            A[k] = R[j]
            j = j + 1
        }
    }
}
fun Sort(A:MutableList<Long>,left:Int, right:Int){
    if (left+1 < right) {
        var mid = (left + right) / 2;
        Sort(A, left, mid)
        Sort(A, mid, right)
        merge(A, left, mid, right)
    }
}
fun gcd(a :Long ,b:Long):Long{
    if (a % b == 0L){
        return b
    }else{
        return gcd(b,a%b)
    }
}
0