結果
問題 |
No.1191 数え上げを愛したい(数列編)
|
ユーザー |
|
提出日時 | 2022-03-17 09:45:59 |
言語 | PyPy3 (7.3.15) |
結果 |
AC
|
実行時間 | 95 ms / 2,000 ms |
コード長 | 730 bytes |
コンパイル時間 | 156 ms |
コンパイル使用メモリ | 82,580 KB |
実行使用メモリ | 85,228 KB |
最終ジャッジ日時 | 2024-10-01 05:11:49 |
合計ジャッジ時間 | 2,950 ms |
ジャッジサーバーID (参考情報) |
judge5 / judge2 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
other | AC * 26 |
ソースコード
N, M, A, B = map(int, input().split()) MOD = 998244353 MAX = 3 * 10 ** 5 + 10 # https://tjkendev.github.io/procon-library/python/math/factorial.html から拝借しています。 fact = [1] * (MAX + 1) rfact = [1] * (MAX + 1) r = 1 for i in range(1, MAX + 1): fact[i] = r = r * i % MOD rfact[MAX] = r = pow(fact[MAX], MOD - 2, MOD) for i in range(MAX, 0, -1): rfact[i - 1] = r = r * i % MOD def comb(n, k, error=0): if not 0 <= k <= n: return error return fact[n] * rfact[k] * rfact[n - k] % MOD ans = 0 for s1 in range(1, M + 1): sn = min(s1 + B, M) d = sn - s1 - A * (N - 1) if d < 0: continue ans += comb(d + N - 1, N - 1) ans %= MOD ans *= fact[N] ans %= MOD print(ans)