結果
問題 |
No.3213 depth max K
|
ユーザー |
|
提出日時 | 2025-07-07 15:54:45 |
言語 | PyPy3 (7.3.15) |
結果 |
AC
|
実行時間 | 92 ms / 2,000 ms |
コード長 | 543 bytes |
コンパイル時間 | 1,148 ms |
コンパイル使用メモリ | 82,464 KB |
実行使用メモリ | 92,384 KB |
最終ジャッジ日時 | 2025-07-07 15:54:52 |
合計ジャッジ時間 | 5,493 ms |
ジャッジサーバーID (参考情報) |
judge2 / judge4 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 3 |
other | AC * 41 |
ソースコード
lim = 1 << 20 mod = 998244353 fac = [1] * lim finv = [1] * lim for i in range(2, lim): fac[i] = fac[i-1] * i % mod finv[lim-1] = pow(fac[lim-1], -1, mod) for i in reversed(range(2, lim)): finv[i-1] = finv[i] * i % mod def C(a, b): if 0 <= b <= a: return fac[a] * finv[b] % mod * finv[a - b] % mod else: return 0 n, k = map(int, input().split()) ans = 0 for i in range(-n, n+1): ans += C(2*n, n+i*(k+2)) - C(2*n, n-k-1+i*(k+2)) ans -= C(2*n, n+i*(k+1)) - C(2*n, n-k+i*(k+1)) ans %= mod print(ans)