結果

問題 No.2944 Sigma Partition Problem
ユーザー PNJPNJ
提出日時 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
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 1,229 ms
449,936 KB
testcase_01 AC 1,231 ms
449,644 KB
testcase_02 WA -
testcase_03 WA -
testcase_04 WA -
testcase_05 WA -
testcase_06 WA -
testcase_07 WA -
testcase_08 WA -
testcase_09 WA -
testcase_10 WA -
testcase_11 WA -
testcase_12 WA -
testcase_13 WA -
testcase_14 WA -
testcase_15 WA -
testcase_16 WA -
testcase_17 WA -
testcase_18 WA -
testcase_19 WA -
testcase_20 WA -
testcase_21 WA -
testcase_22 WA -
testcase_23 AC 1,260 ms
450,128 KB
testcase_24 WA -
testcase_25 WA -
権限があれば一括ダウンロードができます

ソースコード

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