結果

問題 No.2944 Sigma Partition Problem
コンテスト
ユーザー PNJ
提出日時 2024-10-18 21:49:16
言語 PyPy3
(7.3.15)
結果
WA  
実行時間 -
コード長 718 bytes
コンパイル時間 180 ms
コンパイル使用メモリ 82,288 KB
実行使用メモリ 450,624 KB
最終ジャッジ日時 2024-10-18 22:39:04
合計ジャッジ時間 34,637 ms
ジャッジサーバーID
（参考情報） 		judge5 / judge
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 2
other AC * 1 WA * 23
権限があれば一括ダウンロードができます

ソースコード

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

g = [[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):
    g[n][k] = (g[n][k] + g[n][k - 1] + dp[k][n - k]) % mod

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