結果

問題 No.2005 Sum of Power Sums
コンテスト
ユーザー Onikawa Yozy
提出日時 2026-07-30 14:35:07
言語 PyPy3
(7.3.17)
コンパイル:
pypy3 -mpy_compile _filename_
実行:
pypy3 _filename_
結果
AC  
実行時間 807 ms / 2,000 ms
+ 854µs
コード長 606 bytes
記録
記録タグの例:
初AC ショートコード 純ショートコード 純主流ショートコード 最速実行時間
コンパイル時間 233 ms
コンパイル使用メモリ 96,500 KB
実行使用メモリ 109,184 KB
最終ジャッジ日時 2026-07-30 14:35:18
合計ジャッジ時間 10,249 ms
ジャッジサーバーID
（参考情報） 		judge1_0 / judge2_0
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 18
権限があれば一括ダウンロードができます

ソースコード

diff #
raw source code 

m = 998244353
n, m_val = map(int, input().split())
a, b = 1, 1
for i in range(n):
    a = a * (n + m_val - i) % m
    b = b * (i + 1) % m
cur_a = [a, 0]
r = [cur_a]
for i in range(1, 5001):
    nxt_a = []
    for j in range(i + 1):
        t1 = cur_a[j - 1] * (i - j) * (n + m_val + j) % m if j > 0 else 0
        t2 = cur_a[j] * (j + 1) * (m_val + j - i + 1) % m if j < len(cur_a) else 0
        nxt_a.append((t1 + t2) % m)
    nxt_a.append(0)
    cur_a = nxt_a
    sm = sum(cur_a[:-1]) % m
    b = b * (n + i) % m
    r.append((sm * pow(b, -1, m)) % m)
print(sum(r[int(i)] for i in input().split()) % m)
0