結果

問題 No.1191 数え上げを愛したい（数列編）
ユーザー yudedako
提出日時 2020-08-28 18:50:08
言語 Kotlin
(2.1.0)
結果
AC  
実行時間 463 ms / 2,000 ms
コード長 904 bytes
コンパイル時間 14,254 ms
コンパイル使用メモリ 442,020 KB
実行使用メモリ 78,016 KB
最終ジャッジ日時 2024-11-13 23:35:56
合計ジャッジ時間 25,768 ms
ジャッジサーバーID
（参考情報） 		judge3 / judge4
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
other AC * 26
権限があれば一括ダウンロードができます
コンパイルメッセージ
Main.kt:14:24: warning: name shadowed: n
    val combination = {n: Int, r: Int ->
                       ^

ソースコード

diff # 

import kotlin.math.min

fun main() {
    val mod = 998244353
    val (n, m, a, b) = readLine()!!.trim().split(' ').map(String::toInt)
    val factorial = LongArray(m + n + 1){1L}
    val div = LongArray(m + n + 1){1L}
    val inverse = LongArray(m + n + 1){1L}
    for (i in 2 until factorial.size) {
        factorial[i] = i * factorial[i - 1] % mod
        div[i] = (mod - mod / i) * div[mod % i] % mod
        inverse[i] = div[i] * inverse[i - 1] % mod
    }
    val combination = {n: Int, r: Int ->
        if (n >= r) factorial[n] * inverse[r] % mod * inverse[n - r] % mod
        else 0
    }
    val result = (1 .. m).map {smaller ->
        val greater = min(smaller + b, m)
        val diff = greater - a * n.toLong() + a - smaller
        if (diff >= 0)
            combination(diff.toInt() + n - 1, n - 1)
        else
            0
    }
    println(result.sum() % mod * factorial[n] % mod)
}
0