結果

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

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 861 ms
327,204 KB
testcase_01 AC 852 ms
326,928 KB
testcase_02 WA -
testcase_03 WA -
testcase_04 WA -
testcase_05 AC 907 ms
326,796 KB
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 AC 828 ms
326,976 KB
testcase_16 WA -
testcase_17 WA -
testcase_18 WA -
testcase_19 WA -
testcase_20 WA -
testcase_21 WA -
testcase_22 WA -
testcase_23 AC 885 ms
326,976 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

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