結果

問題 No.3538 Not First Place
コンテスト
ユーザー kidodesu
提出日時 2026-05-09 00:12:49
言語 PyPy3
(7.3.17)
コンパイル:
pypy3 -mpy_compile _filename_
実行:
pypy3 _filename_
結果
AC  
実行時間 2,922 ms / 5,000 ms
コード長 849 bytes
記録
記録タグの例:
初AC ショートコード 純ショートコード 純主流ショートコード 最速実行時間
コンパイル時間 185 ms
コンパイル使用メモリ 85,536 KB
実行使用メモリ 237,304 KB
最終ジャッジ日時 2026-05-09 00:13:11
合計ジャッジ時間 17,398 ms
ジャッジサーバーID
(参考情報)
judge2_1 / judge3_0
このコードへのチャレンジ
(要ログイン)
ファイルパターン 結果
sample AC * 2
other AC * 26
権限があれば一括ダウンロードができます

ソースコード

diff #
raw source code

mod = 998244353
comb_num = 10**7 + 2
F = [1] * comb_num
F_ = [1] * comb_num
for i in range(2, comb_num): F[i] = F[i-1]*i%mod
F_[-1] = pow(F[-1], -1, mod)
for i in range(comb_num-1, 0, -1): F_[i-1] = F_[i]*i%mod
def comb(a, b):
    if 0 <= b <= a: return F[a] * F_[b] * F_[a-b] % mod
    else: return 0

n, m, k, l = list(map(int, input().split()))

ans = 0
def main(box, ball, v):
    rep = 0
    cnt = 0
    while cnt * v <= ball and cnt <= box:
        n_ball = ball - cnt * v
        rep = (rep + (1 if not cnt % 2 else -1) * comb(n_ball+box-1, box-1) * comb(box, cnt)) % mod
        #print(n_ball, cnt, comb(n_ball+box-1, box-1), comb(box, cnt))
        cnt += 1
    return rep

for v1 in range(l, m):
    ans = (ans + main(n-1, m*k-v1, m+1) - main(n-1, m*k-v1, v1+1)) % mod
    #print(main(n-1, m*k-v1, m+1), main(n-1, m*k-v1, v1+1))
print(ans)
0