結果

問題 No.1191 数え上げを愛したい（数列編）
コンテスト
ユーザー yudedako
提出日時 2020-08-28 18:47:15
言語 Kotlin
(2.1.0)
結果
RE  
実行時間 -
コード長 843 bytes
コンパイル時間 12,718 ms
コンパイル使用メモリ 443,612 KB
実行使用メモリ 80,272 KB
最終ジャッジ日時 2024-11-13 23:35:26
合計ジャッジ時間 24,783 ms
ジャッジサーバーID
（参考情報） 		judge2 / judge3
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
other AC * 25 RE * 1
権限があれば一括ダウンロードができます
コンパイルメッセージ
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 && r >= 0) 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 + a - smaller
        combination(diff + n - 1, n - 1)
    }
    println(result.sum() % mod * factorial[n] % mod)
}
0