結果

問題 No.2005 Sum of Power Sums
コンテスト
ユーザー 2251799813685248
提出日時 2026-10-23 19:54:53
言語 PyPy3
(8.0.0 + ACL)
コンパイル:
pypy3 -mpy_compile _filename_
実行:
pypy3 _filename_
結果
AC  
実行時間 775 ms / 2,000 ms
+ 140µs
コード長 840 bytes
記録
記録タグの例:
初AC ショートコード 純ショートコード 純主流ショートコード 最速実行時間
コンパイル時間 378 ms
コンパイル使用メモリ 64,448 KB
実行使用メモリ 281,624 KB
最終ジャッジ日時 2026-10-23 19:55:14
合計ジャッジ時間 19,449 ms
ジャッジサーバーID
（参考情報） 		judge1_0 / judge5_0
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 18
権限があれば一括ダウンロードができます

ソースコード

diff #
raw source code 

N,M = map(int, input().split())
a = [0]*5001
nSk = [[0]*5005 for i in range(5005)]
nSk[0][0] = 1
for i in range(1,5005):
  for j in range(1, 5005):
    nSk[i][j] = nSk[i-1][j-1] + j*nSk[i-1][j]
    nSk[i][j] %= 998244353
nCr = [0]*5001
fact = [1]*5001
temp = 1
for i in range(1,N+1):
  temp *= (M+i)*pow(i,-1,998244353)%998244353
  temp %= 998244353
for i in range(5001):
  nCr[i] = temp
  temp *= (M+N+i+1)*pow(N+i+1,-1,998244353)%998244353
  temp %= 998244353
for i in range(1,5001):
  fact[i] = i*fact[i-1]%998244353
for k in range(5001):
  for j in range(5001):
    a[k] += nCr[j]*(1 if (k+j)%2 == 0 else 998244352)%998244353*fact[j]%998244353*nSk[k+1][j+1]%998244353
    if a[k] >= 998244353:
      a[k] -= 998244353
ans = 0
for k in list(map(int, input().split())):
  ans += a[k]
  if ans >= 998244353:
    ans -= 998244353
print(ans)
0