結果

問題 No.1191 数え上げを愛したい(数列編)
ユーザー yudedakoyudedako
提出日時 2020-08-28 18:45:54
言語 Kotlin
(1.9.23)
結果
RE  
実行時間 -
コード長 833 bytes
コンパイル時間 15,024 ms
コンパイル使用メモリ 439,348 KB
実行使用メモリ 75,004 KB
最終ジャッジ日時 2024-04-26 12:24:41
合計ジャッジ時間 25,485 ms
ジャッジサーバーID
(参考情報)
judge4 / judge1
このコードへのチャレンジ
(要ログイン)

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 443 ms
64,744 KB
testcase_01 AC 435 ms
72,348 KB
testcase_02 AC 450 ms
72,812 KB
testcase_03 AC 428 ms
71,680 KB
testcase_04 AC 467 ms
75,004 KB
testcase_05 AC 419 ms
63,008 KB
testcase_06 AC 421 ms
73,532 KB
testcase_07 AC 450 ms
74,092 KB
testcase_08 AC 481 ms
74,152 KB
testcase_09 AC 470 ms
73,960 KB
testcase_10 AC 446 ms
71,460 KB
testcase_11 AC 446 ms
71,308 KB
testcase_12 AC 439 ms
73,136 KB
testcase_13 AC 445 ms
73,036 KB
testcase_14 AC 445 ms
74,660 KB
testcase_15 AC 381 ms
57,252 KB
testcase_16 AC 385 ms
57,272 KB
testcase_17 AC 381 ms
56,516 KB
testcase_18 AC 366 ms
56,468 KB
testcase_19 AC 376 ms
58,164 KB
testcase_20 RE -
testcase_21 AC 353 ms
54,772 KB
testcase_22 AC 421 ms
64,984 KB
testcase_23 AC 304 ms
51,844 KB
testcase_24 AC 304 ms
51,840 KB
testcase_25 AC 303 ms
51,780 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 + a - smaller
        combination(diff + n - 1, n - 1)
    }
    println(result.sum() % mod * factorial[n] % mod)
}
0