結果
問題 | No.1772 Many DELETEQs |
ユーザー | 👑 SPD_9X2 |
提出日時 | 2021-11-23 18:37:02 |
言語 | PyPy3 (7.3.15) |
結果 |
AC
|
実行時間 | 1,088 ms / 3,500 ms |
コード長 | 1,545 bytes |
コンパイル時間 | 596 ms |
コンパイル使用メモリ | 86,992 KB |
実行使用メモリ | 339,312 KB |
最終ジャッジ日時 | 2023-09-07 21:58:26 |
合計ジャッジ時間 | 12,448 ms |
ジャッジサーバーID (参考情報) |
judge12 / judge14 |
(要ログイン)
テストケース
テストケース表示入力 | 結果 | 実行時間 実行使用メモリ |
---|---|---|
testcase_00 | AC | 995 ms
328,032 KB |
testcase_01 | AC | 972 ms
328,412 KB |
testcase_02 | AC | 985 ms
328,284 KB |
testcase_03 | AC | 1,087 ms
339,024 KB |
testcase_04 | AC | 1,088 ms
339,312 KB |
testcase_05 | AC | 1,083 ms
339,128 KB |
testcase_06 | AC | 1,047 ms
339,296 KB |
ソースコード
""" Many Deleteqs 想定解v1 dp[i][j] = ABをi個、BAをj個最小で使うことで作成可能な文字列の個数 AB,BAは対消滅可能なので、 i <= x , j <= y , i-j = x-y となる i,jに関して総和を求めればよい。 x-y >= 0 となるように、x,yを交換する。 iを決めると、jも決まる。 ここから、iの範囲が決まる。 後は、x-yごとにindexをiとして累積和を取っておくとクエリO(1) dp2[a][b] = (i-j==a) かつ、i <= b の場合の dp[i][j] の総和 """ import sys from sys import stdin mod = 998244353 dp = [[0] * 4001 for i in range(4001)] dp[0][0] = 1 for i in range(4001): for j in range(4001): if i != 0 and j != 0: #add AA or BB dp[i][j] += 2 * dp[i-1][j-1] if i != 0: #add AB dp[i][j] += dp[i-1][j] if j != 0: dp[i][j] += dp[i][j-1] dp[i][j] %= mod dp2 = [ [0] * (4002) for i in range(4002) ] for i in range(4001): for j in range(i+1): a = i - j dp2[a][i] += dp[i][j] dp2[a][i] %= mod #累積和 for a in range(4001): for b in range(4000): dp2[a][b+1] += dp2[a][b] dp2[a][b+1] %= mod Q = int(stdin.readline()) ANS = [] for loop in range(Q): x,y = map(int,stdin.readline().split()) #制約チェック assert 1 <= x <= 4000 assert 1 <= y <= 4000 if x < y: x,y = y,x a = x-y ilim = min(x , y+a) ANS.append( dp2[a][ilim] ) print ("\n".join(map(str,ANS)))