結果

問題 No.2944 Sigma Partition Problem
コンテスト
ユーザー PNJ
提出日時 2024-10-18 22:55:28
言語 PyPy3
(7.3.15)
結果
WA  
実行時間 -
コード長 553 bytes
コンパイル時間 194 ms
コンパイル使用メモリ 82,364 KB
実行使用メモリ 327,804 KB
最終ジャッジ日時 2024-10-18 22:57:15
合計ジャッジ時間 24,265 ms
ジャッジサーバーID
（参考情報） 		judge1 / judge
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 2
other AC * 3 WA * 21
権限があれば一括ダウンロードができます

ソースコード

diff # 

M,mod = 4000,998244353
dp = [[0 for i in range(M + 1)] for j in range(M + 1)]
dp[0][0] = 1
for i in range(1,M + 1):
  for j in range(M + 1):
    dp[i][j] = (dp[i][j] + dp[i - 1][j]) % mod
    if j + i <= M:
      dp[i][j + i] = (dp[i][j + i] + dp[i][j]) % mod

f = [[0 for i in range(M + 1)] for j in range(M + 1)]
for n in range(1,M + 1):
  for k in range(1,n + 1):
    f[n][k] = (f[n][k] + f[n][k - 1] + dp[k][n - k]) % mod

for _ in range(int(input())):
  q,n,k = map(int,input().split())
  ans = dp[k][n]
  if q == 2:
    ans = f[n][k]
  print(ans)
0