結果

問題 No.1191 数え上げを愛したい（数列編）
ユーザー yudedakoyudedako
提出日時 2020-08-28 18:50:08
言語 Kotlin
(1.9.23)
結果
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
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 437 ms
67,584 KB
testcase_01 AC 448 ms
73,816 KB
testcase_02 AC 449 ms
73,908 KB
testcase_03 AC 438 ms
76,008 KB
testcase_04 AC 452 ms
78,016 KB
testcase_05 AC 410 ms
65,604 KB
testcase_06 AC 440 ms
77,828 KB
testcase_07 AC 457 ms
77,908 KB
testcase_08 AC 451 ms
77,936 KB
testcase_09 AC 453 ms
77,908 KB
testcase_10 AC 425 ms
71,632 KB
testcase_11 AC 427 ms
71,872 KB
testcase_12 AC 443 ms
77,740 KB
testcase_13 AC 445 ms
77,976 KB
testcase_14 AC 463 ms
77,872 KB
testcase_15 AC 391 ms
61,488 KB
testcase_16 AC 388 ms
61,400 KB
testcase_17 AC 386 ms
59,216 KB
testcase_18 AC 383 ms
59,344 KB
testcase_19 AC 396 ms
61,304 KB
testcase_20 AC 388 ms
61,616 KB
testcase_21 AC 361 ms
59,392 KB
testcase_22 AC 429 ms
67,720 KB
testcase_23 AC 319 ms
56,924 KB
testcase_24 AC 323 ms
57,032 KB
testcase_25 AC 322 ms
56,980 KB
権限があれば一括ダウンロードができます
コンパイルメッセージ
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